Home
last modified time | relevance | path

Searched refs:ADD_VARIANT_BINARY_OP (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dvariant_op_registry_test.cc99 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 …]
Dvariant_op_registry.cc233 REGISTER_UNARY_VARIANT_BINARY_OP_FUNCTION(ADD_VARIANT_BINARY_OP, DEVICE_CPU, \
Dvariant_op_registry.h49 ADD_VARIANT_BINARY_OP = 1, enumerator
/external/tensorflow/tensorflow/core/kernels/data/
Doptional_ops.cu.cc30 REGISTER_UNARY_VARIANT_BINARY_OP_FUNCTION(ADD_VARIANT_BINARY_OP, DEVICE_GPU,
Doptional_ops.cc189 REGISTER_UNARY_VARIANT_BINARY_OP_FUNCTION(ADD_VARIANT_BINARY_OP, DEVICE_CPU,
/external/tensorflow/tensorflow/core/kernels/
Daggregate_ops.cc190 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()
Dlist_kernels.cu.cc124 REGISTER_UNARY_VARIANT_BINARY_OP_FUNCTION(ADD_VARIANT_BINARY_OP, DEVICE_GPU,
Dlist_kernels.cc682 REGISTER_UNARY_VARIANT_BINARY_OP_FUNCTION(ADD_VARIANT_BINARY_OP, DEVICE_CPU,
/external/tensorflow/tensorflow/core/util/
Dtensor_ops_util.h115 ctx, ADD_VARIANT_BINARY_OP, a.scalar<Variant>()(), in BinaryAddTensors()