Home
last modified time | relevance | path

Searched refs:BinaryOpVariants (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dvariant_op_registry_test.cc261 Status s0 = BinaryOpVariants<CPUDevice>( in TEST()
268 TF_EXPECT_OK(BinaryOpVariants<CPUDevice>( in TEST()
288 Status s0 = BinaryOpVariants<GPUDevice>( in TEST()
295 TF_EXPECT_OK(BinaryOpVariants<GPUDevice>( in TEST()
Dvariant_op_registry.h288 Status BinaryOpVariants(OpKernelContext* ctx, VariantBinaryOp op, in BinaryOpVariants() function
/external/tensorflow/tensorflow/core/kernels/
Daggregate_ops.cc189 ctx, BinaryOpVariants<Device>( in Compute()
195 OP_REQUIRES_OK(ctx, BinaryOpVariants<Device>(ctx, ADD_VARIANT_BINARY_OP, in Compute()
/external/tensorflow/tensorflow/core/util/
Dtensor_ops_util.h114 TF_RETURN_IF_ERROR(BinaryOpVariants<Device>( in BinaryAddTensors()