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

Fix missing status check.

Browse Source 
PiperOrigin-RevId: 322479633
Change-Id: I6452af8b6f106d6681307caab1c9b41e514d4694
This commit is contained in:
Russell Power 2020-07-21 17:48:06 -07:00 committed by TensorFlower Gardener
parent d8bbfa5d53
commit 2df63a4fa1
2 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/core/tpu/tpu_library_init_fns.inc
View File

@ -1,4 +1,5 @@
#include "third_party/tensorflow/stream_executor/tpu/tpu_executor_c_api.h"
namespace {
tensorflow::Status SetTpuConfigStructFns(void* library_handle) {

3
tensorflow/core/tpu/tpu_node_device.cc
View File

@ -90,6 +90,9 @@ Status TpuPaddedShapeFn(const Tensor& tensor, xla::Shape* shape) {
ApiConverter::StackHelper<XLA_Shape> tpu_shape;
tpu::ExecutorApiFn()->XlaShapeToTpuPaddedShapeFn(
&se_shape.value, &tpu_shape.value, status.c_status);
if (!status.ok()) {
return status.status();
}
*shape = tpu_shape.AsCpp<xla::Shape>();
return Status::OK();
}