Home
last modified time | relevance | path

Searched refs:kDeviceRetOp (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfunction_ops.h27 static const char* const kDeviceRetOp = FunctionLibraryDefinition::kDeviceRetOp; variable
Dfunction_ops.cc74 REGISTER_SYSTEM_KERNEL_BUILDER(Name(kDeviceRetOp).Device(DEVICE_CPU), RetvalOp);
140 Name(kDeviceRetOp).Device(DEVICE_GPU).TypeConstraint<int32>("T"), RetvalOp);
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_ops.h228 Name(kDeviceRetOp).Device(DEVICE).TypeConstraint<int32>("T"), RetvalOp); \
/external/tensorflow/tensorflow/core/framework/
Dfunction.h402 static constexpr const char* const kDeviceRetOp = "_DeviceRetval"; variable
Dfunction.cc341 gnode->set_op(FunctionLibraryDefinition::kDeviceRetOp); in AddReturnNode()
591 n->op() == FunctionLibraryDefinition::kDeviceRetOp) { in Print()
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction.cc54 static constexpr const char* const kDeviceRetOp = variable
55 FunctionLibraryDefinition::kDeviceRetOp;
1930 if (n->type_string() == kRetOp || n->type_string() == kDeviceRetOp) { in FunctionBody()