Searched refs:ZEROS_LIKE_VARIANT_UNARY_OP (Results 1 – 12 of 12) 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, 198 ZEROS_LIKE_VARIANT_UNARY_OP, DEVICE_CPU, TypeIndex::Make<Blah>()), in TEST() 207 ZEROS_LIKE_VARIANT_UNARY_OP, v, &v_out); in TEST() 214 null_context_pointer, ZEROS_LIKE_VARIANT_UNARY_OP, v, &v_out)); in TEST() 224 ZEROS_LIKE_VARIANT_UNARY_OP, DEVICE_GPU, TypeIndex::Make<Blah>()), in TEST() 233 ZEROS_LIKE_VARIANT_UNARY_OP, v, &v_out); in TEST() 240 null_context_pointer, ZEROS_LIKE_VARIANT_UNARY_OP, v, &v_out)); in TEST() 252 registry.RegisterUnaryOpFn(ZEROS_LIKE_VARIANT_UNARY_OP, DEVICE_CPU, in TEST() 254 EXPECT_DEATH(registry.RegisterUnaryOpFn(ZEROS_LIKE_VARIANT_UNARY_OP, in TEST() [all …]
|
D | variant_op_registry.cc | 33 case ZEROS_LIKE_VARIANT_UNARY_OP: in VariantUnaryOpToString() 175 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 | 188 REGISTER_UNARY_VARIANT_UNARY_OP_FUNCTION(ZEROS_LIKE_VARIANT_UNARY_OP,
|
/external/tensorflow/tensorflow/core/kernels/ |
D | map_kernels.cc | 55 REGISTER_UNARY_VARIANT_UNARY_OP_FUNCTION(ZEROS_LIKE_VARIANT_UNARY_OP,
|
D | ragged_tensor_variant.cc | 67 ZEROS_LIKE_VARIANT_UNARY_OP, DEVICE_CPU, RaggedTensorVariant,
|
D | list_kernels.cu.cc | 124 REGISTER_UNARY_VARIANT_UNARY_OP_FUNCTION(ZEROS_LIKE_VARIANT_UNARY_OP,
|
D | constant_op.cc | 245 ctx, ZEROS_LIKE_VARIANT_UNARY_OP, v, out_v)); in Compute()
|
D | list_kernels.cc | 620 REGISTER_UNARY_VARIANT_UNARY_OP_FUNCTION(ZEROS_LIKE_VARIANT_UNARY_OP,
|
/external/tensorflow/tensorflow/core/kernels/sparse/ |
D | zeros_op.cc | 87 ZEROS_LIKE_VARIANT_UNARY_OP, DEVICE_GPU, CSRSparseMatrix,
|
/external/tensorflow/tensorflow/core/util/ |
D | tensor_ops_util.h | 59 UnaryOpVariant<Device>(ctx, ZEROS_LIKE_VARIANT_UNARY_OP, in ZerosLikeTensor()
|