Searched refs:ADD_VARIANT_BINARY_OP (Results 1 – 9 of 9) 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, 251 ADD_VARIANT_BINARY_OP, DEVICE_CPU, MakeTypeIndex<Blah>()), in TEST() 262 null_context_pointer, ADD_VARIANT_BINARY_OP, v_a, v_b, &v_out); in TEST() 269 null_context_pointer, ADD_VARIANT_BINARY_OP, v_a, v_b, &v_out)); in TEST() 278 ADD_VARIANT_BINARY_OP, DEVICE_GPU, MakeTypeIndex<Blah>()), in TEST() 289 null_context_pointer, ADD_VARIANT_BINARY_OP, v_a, v_b, &v_out); in TEST() 296 null_context_pointer, ADD_VARIANT_BINARY_OP, v_a, v_b, &v_out)); in TEST() 308 registry.RegisterBinaryOpFn(ADD_VARIANT_BINARY_OP, DEVICE_CPU, kTypeIndex, f); in TEST() 309 EXPECT_DEATH(registry.RegisterBinaryOpFn(ADD_VARIANT_BINARY_OP, DEVICE_CPU, in TEST() [all …]
|
D | variant_op_registry.cc | 233 REGISTER_UNARY_VARIANT_BINARY_OP_FUNCTION(ADD_VARIANT_BINARY_OP, DEVICE_CPU, \
|
D | variant_op_registry.h | 49 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 | 189 REGISTER_UNARY_VARIANT_BINARY_OP_FUNCTION(ADD_VARIANT_BINARY_OP, DEVICE_CPU,
|
/external/tensorflow/tensorflow/core/kernels/ |
D | aggregate_ops.cc | 190 ctx, ADD_VARIANT_BINARY_OP, ctx->input(0).scalar<Variant>()(), in Compute() 195 OP_REQUIRES_OK(ctx, BinaryOpVariants<Device>(ctx, ADD_VARIANT_BINARY_OP, in Compute()
|
D | list_kernels.cu.cc | 124 REGISTER_UNARY_VARIANT_BINARY_OP_FUNCTION(ADD_VARIANT_BINARY_OP, DEVICE_GPU,
|
D | list_kernels.cc | 682 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()
|