Searched refs:ZEROS_LIKE_VARIANT_UNARY_OP (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | variant_op_registry_test.cc | 91 REGISTER_UNARY_VARIANT_UNARY_OP_FUNCTION(ZEROS_LIKE_VARIANT_UNARY_OP,
95 REGISTER_UNARY_VARIANT_UNARY_OP_FUNCTION(ZEROS_LIKE_VARIANT_UNARY_OP,
180 ZEROS_LIKE_VARIANT_UNARY_OP, DEVICE_CPU, MakeTypeIndex<Blah>()), in TEST()
189 ZEROS_LIKE_VARIANT_UNARY_OP, v, &v_out); in TEST()
197 null_context_pointer, ZEROS_LIKE_VARIANT_UNARY_OP, v, &v_out)); in TEST()
206 ZEROS_LIKE_VARIANT_UNARY_OP, DEVICE_GPU, MakeTypeIndex<Blah>()), in TEST()
215 ZEROS_LIKE_VARIANT_UNARY_OP, v, &v_out); in TEST()
223 null_context_pointer, ZEROS_LIKE_VARIANT_UNARY_OP, v, &v_out)); in TEST()
235 registry.RegisterUnaryOpFn(ZEROS_LIKE_VARIANT_UNARY_OP, DEVICE_CPU, in TEST()
237 EXPECT_DEATH(registry.RegisterUnaryOpFn(ZEROS_LIKE_VARIANT_UNARY_OP, in TEST()
[all …]
|
| D | variant_op_registry.cc | 190 REGISTER_UNARY_VARIANT_UNARY_OP_FUNCTION(ZEROS_LIKE_VARIANT_UNARY_OP, \
|
| D | variant_op_registry.h | 43 ZEROS_LIKE_VARIANT_UNARY_OP = 1, enumerator
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | optional_ops.cu.cc | 26 REGISTER_UNARY_VARIANT_UNARY_OP_FUNCTION(ZEROS_LIKE_VARIANT_UNARY_OP,
|
| D | optional_ops.cc | 185 REGISTER_UNARY_VARIANT_UNARY_OP_FUNCTION(ZEROS_LIKE_VARIANT_UNARY_OP,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | list_kernels.cu.cc | 127 REGISTER_UNARY_VARIANT_UNARY_OP_FUNCTION(ZEROS_LIKE_VARIANT_UNARY_OP,
|
| D | constant_op.cc | 269 ctx, ZEROS_LIKE_VARIANT_UNARY_OP, v, out_v)); in Compute()
|
| D | list_kernels.cc | 686 REGISTER_UNARY_VARIANT_UNARY_OP_FUNCTION(ZEROS_LIKE_VARIANT_UNARY_OP,
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_ops_util.h | 59 UnaryOpVariant<Device>(ctx, ZEROS_LIKE_VARIANT_UNARY_OP, in ZerosLikeTensor()
|