Home
last modified time | relevance | path

Searched refs:ZEROS_LIKE_VARIANT_UNARY_OP (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dvariant_op_registry_test.cc91 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 …]
Dvariant_op_registry.cc33 case ZEROS_LIKE_VARIANT_UNARY_OP: in VariantUnaryOpToString()
175 REGISTER_UNARY_VARIANT_UNARY_OP_FUNCTION(ZEROS_LIKE_VARIANT_UNARY_OP, \
Dvariant_op_registry.h43 ZEROS_LIKE_VARIANT_UNARY_OP = 1, enumerator
/external/tensorflow/tensorflow/core/kernels/data/
Doptional_ops.cu.cc26 REGISTER_UNARY_VARIANT_UNARY_OP_FUNCTION(ZEROS_LIKE_VARIANT_UNARY_OP,
Doptional_ops.cc188 REGISTER_UNARY_VARIANT_UNARY_OP_FUNCTION(ZEROS_LIKE_VARIANT_UNARY_OP,
/external/tensorflow/tensorflow/core/kernels/
Dmap_kernels.cc55 REGISTER_UNARY_VARIANT_UNARY_OP_FUNCTION(ZEROS_LIKE_VARIANT_UNARY_OP,
Dragged_tensor_variant.cc67 ZEROS_LIKE_VARIANT_UNARY_OP, DEVICE_CPU, RaggedTensorVariant,
Dlist_kernels.cu.cc124 REGISTER_UNARY_VARIANT_UNARY_OP_FUNCTION(ZEROS_LIKE_VARIANT_UNARY_OP,
Dconstant_op.cc245 ctx, ZEROS_LIKE_VARIANT_UNARY_OP, v, out_v)); in Compute()
Dlist_kernels.cc620 REGISTER_UNARY_VARIANT_UNARY_OP_FUNCTION(ZEROS_LIKE_VARIANT_UNARY_OP,
/external/tensorflow/tensorflow/core/kernels/sparse/
Dzeros_op.cc87 ZEROS_LIKE_VARIANT_UNARY_OP, DEVICE_GPU, CSRSparseMatrix,
/external/tensorflow/tensorflow/core/util/
Dtensor_ops_util.h59 UnaryOpVariant<Device>(ctx, ZEROS_LIKE_VARIANT_UNARY_OP, in ZerosLikeTensor()