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

[XLA] Unbreak the build by using protobuf's int64 type.

Browse Source 
PiperOrigin-RevId: 290868470
Change-Id: I35f10bd922e458252236835069408699a4c68da9
This commit is contained in:
Tim Shen 2020-01-21 18:30:40 -08:00 committed by TensorFlower Gardener
parent 73ac5eb902
commit 933fa7cfeb
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/compiler/xla/service/mlir_gpu/experimental/conv_emitter/conv_emitter.cc
View File

@ -58,9 +58,10 @@ struct ShapeInfo {
mlir::Type element_type;
};
ShapeInfo GetShapeInfo(const Shape& shape, int64 n_dim, int64 c_dim,
absl::Span<const int64> spatial_dims,
mlir::Builder builder) {
ShapeInfo GetShapeInfo(
const Shape& shape, int64 n_dim, int64 c_dim,
absl::Span<const tensorflow::protobuf_int64> spatial_dims,
mlir::Builder builder) {
ShapeInfo shape_info;
std::vector<int64> physical_to_logical(