Searched refs:UnaryOpVariant (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/framework/ |
| D | variant_op_registry_test.cc | 206 Status s0 = UnaryOpVariant<CPUDevice>(null_context_pointer, in TEST()
213 TF_EXPECT_OK(UnaryOpVariant<CPUDevice>( in TEST()
232 Status s0 = UnaryOpVariant<GPUDevice>(null_context_pointer, in TEST()
239 TF_EXPECT_OK(UnaryOpVariant<GPUDevice>( in TEST()
|
| D | variant_op_registry.h | 312 Status UnaryOpVariant(OpKernelContext* ctx, VariantUnaryOp op, const Variant& v, in UnaryOpVariant() function
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_ops_util.h | 59 UnaryOpVariant<Device>(ctx, ZEROS_LIKE_VARIANT_UNARY_OP, in ZerosLikeTensor()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | constant_op.cc | 244 OP_REQUIRES_OK(ctx, UnaryOpVariant<Device>( in Compute()
|
| D | cwise_ops_common.h | 327 OP_REQUIRES_OK(ctx, UnaryOpVariant<Device>(ctx, OpEnum, v, &v_out)); in Compute()
|