experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #45275 from nouiz:upstream_nvbug3180567

Browse Source 
PiperOrigin-RevId: 348090921
Change-Id: I01f5f73f16e458f0f6ec6177133f66cbdc41f572
This commit is contained in:
TensorFlower Gardener 2020-12-17 14:13:34 -08:00
commit 8a894275a4
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/tf2xla/kernels/conv_op_helpers.cc
View File

@ -179,7 +179,7 @@ Status ConvBackpropComputeDimensionsV2XlaShapes(
} // anonymous namespace
absl::Span<const DataType> GetXlaConvTypes() {
std::vector<DataType> GetXlaConvTypes() {
return {DT_FLOAT, DT_BFLOAT16, DT_HALF, DT_DOUBLE};
}

2
tensorflow/compiler/tf2xla/kernels/conv_op_helpers.h
View File

@ -37,7 +37,7 @@ namespace tensorflow {
// We don't support integers for convolutions, so we list the supported types
// here.
absl::Span<const DataType> GetXlaConvTypes();
std::vector<DataType> GetXlaConvTypes();
// ConvOpAttrs contains all of the metadata necessary to specify a TF or XLA
// convolution.