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

Sanitize with clang-format

Browse Source 
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
This commit is contained in:
Yong Tang 2020-04-14 16:35:30 +00:00
parent da3c1f4b01
commit 2ef79b9c91
1 changed files with 7 additions and 2 deletions

9
tensorflow/compiler/tf2xla/kernels/ensure_shape_op.cc
View File

@ -38,12 +38,17 @@ class EnsureShapeOp : public XlaOpKernel {
const TensorShape shape = ctx->InputShape(0);
// valiate shape
OP_REQUIRES(ctx, expected_shape_.IsCompatibleWith(shape),
errors::InvalidArgument("Shape of tensor ", this->def().input(0), " ", shape.DebugString(), " is not compatible with expected shape ", expected_shape_.DebugString(), "."));
OP_REQUIRES(
ctx, expected_shape_.IsCompatibleWith(shape),
errors::InvalidArgument("Shape of tensor ", this->def().input(0), " ",
shape.DebugString(),
" is not compatible with expected shape ",
expected_shape_.DebugString(), "."));
// If shape matches, outputs the tensor.
ctx->SetOutput(0, ctx->Input(0));
}
private:
PartialTensorShape expected_shape_;
};