diff options
Diffstat (limited to 'tensorflow/core/kernels/cast_op_test.cc')
-rw-r--r-- | tensorflow/core/kernels/cast_op_test.cc | 15 |
1 files changed, 8 insertions, 7 deletions
diff --git a/tensorflow/core/kernels/cast_op_test.cc b/tensorflow/core/kernels/cast_op_test.cc index ffad7fd02e..5b7529bb8a 100644 --- a/tensorflow/core/kernels/cast_op_test.cc +++ b/tensorflow/core/kernels/cast_op_test.cc @@ -49,17 +49,18 @@ class CastOpTest : public OpsTestBase { TF_EXPECT_OK(InitOp()); } - template <typename IN, typename OUT> + template <typename INPUT, typename OUTPUT> void CheckCast() { - DataType in_type = DataTypeToEnum<IN>::v(); - DataType out_type = DataTypeToEnum<OUT>::v(); + DataType in_type = DataTypeToEnum<INPUT>::v(); + DataType out_type = DataTypeToEnum<OUTPUT>::v(); MakeOp(in_type, out_type); - AddInputFromArray<IN>(TensorShape({1, 2, 2, 1}), - {IN(1), IN(2), IN(3), IN(4)}); + AddInputFromArray<INPUT>(TensorShape({1, 2, 2, 1}), + {INPUT(1), INPUT(2), INPUT(3), INPUT(4)}); TF_ASSERT_OK(RunOpKernel()); Tensor expected(allocator(), out_type, TensorShape({1, 2, 2, 1})); - test::FillValues<OUT>(&expected, {OUT(1), OUT(2), OUT(3), OUT(4)}); - test::ExpectTensorEqual<OUT>(expected, *GetOutput(0)); + test::FillValues<OUTPUT>(&expected, + {OUTPUT(1), OUTPUT(2), OUTPUT(3), OUTPUT(4)}); + test::ExpectTensorEqual<OUTPUT>(expected, *GetOutput(0)); } }; |