Searched refs:ADD_VARIANT_BINARY_OP (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | variant_op_registry_test.cc | 99 REGISTER_UNARY_VARIANT_BINARY_OP_FUNCTION(ADD_VARIANT_BINARY_OP, DEVICE_CPU,
102 REGISTER_UNARY_VARIANT_BINARY_OP_FUNCTION(ADD_VARIANT_BINARY_OP, DEVICE_GPU,
268 ADD_VARIANT_BINARY_OP, DEVICE_CPU, TypeIndex::Make<Blah>()), in TEST()
279 null_context_pointer, ADD_VARIANT_BINARY_OP, v_a, v_b, &v_out); in TEST()
286 null_context_pointer, ADD_VARIANT_BINARY_OP, v_a, v_b, &v_out)); in TEST()
295 ADD_VARIANT_BINARY_OP, DEVICE_GPU, TypeIndex::Make<Blah>()), in TEST()
306 null_context_pointer, ADD_VARIANT_BINARY_OP, v_a, v_b, &v_out); in TEST()
313 null_context_pointer, ADD_VARIANT_BINARY_OP, v_a, v_b, &v_out)); in TEST()
325 registry.RegisterBinaryOpFn(ADD_VARIANT_BINARY_OP, DEVICE_CPU, kTypeIndex, f); in TEST()
326 EXPECT_DEATH(registry.RegisterBinaryOpFn(ADD_VARIANT_BINARY_OP, DEVICE_CPU, in TEST()
[all …]
|
| D | variant_op_registry.cc | 44 case ADD_VARIANT_BINARY_OP: in VariantBinaryOpToString()
197 REGISTER_UNARY_VARIANT_BINARY_OP_FUNCTION(ADD_VARIANT_BINARY_OP, DEVICE_CPU, \
|
| D | variant_op_registry.h | 51 ADD_VARIANT_BINARY_OP = 1, enumerator
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | optional_ops.cu.cc | 30 REGISTER_UNARY_VARIANT_BINARY_OP_FUNCTION(ADD_VARIANT_BINARY_OP, DEVICE_GPU,
|
| D | optional_ops.cc | 192 REGISTER_UNARY_VARIANT_BINARY_OP_FUNCTION(ADD_VARIANT_BINARY_OP, DEVICE_CPU,
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | map_kernels.cc | 51 REGISTER_UNARY_VARIANT_BINARY_OP_FUNCTION(ADD_VARIANT_BINARY_OP, DEVICE_CPU,
|
| D | ragged_tensor_variant.cc | 71 ADD_VARIANT_BINARY_OP, DEVICE_CPU, RaggedTensorVariant,
|
| D | list_kernels.cu.cc | 121 REGISTER_UNARY_VARIANT_BINARY_OP_FUNCTION(ADD_VARIANT_BINARY_OP, DEVICE_GPU,
|
| D | aggregate_ops.cc | 244 BinaryOpVariants<Device>(ctx, ADD_VARIANT_BINARY_OP, a, b, c)); in AddVariantTo()
|
| D | list_kernels.cc | 616 REGISTER_UNARY_VARIANT_BINARY_OP_FUNCTION(ADD_VARIANT_BINARY_OP, DEVICE_CPU,
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_ops_util.h | 115 ctx, ADD_VARIANT_BINARY_OP, a.scalar<Variant>()(), in BinaryAddTensors()
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | add_op.cc | 264 ADD_VARIANT_BINARY_OP, DEVICE_GPU, CSRSparseMatrix,
|