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

Fix a bug. The Span returned point to an initializer list that is removed. This isn't allowed.

Browse Source
This commit is contained in:
Frederic Bastien 2020-11-30 08:36:47 -08:00
parent 634864312d
commit a5a5abfe75
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.