Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dfunction_ops.h27 static const char* const kDeviceRetOp = FunctionLibraryDefinition::kDeviceRetOp; variable
Dfunction_ops.cc90 REGISTER_SYSTEM_KERNEL_BUILDER(Name(kDeviceRetOp).Device(DEVICE_CPU), RetvalOp);
139 Name(kDeviceRetOp).Device(DEVICE_GPU).TypeConstraint<int32>("T"), RetvalOp);
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction_body.cc35 n->type_string() == FunctionLibraryDefinition::kDeviceRetOp) { in FunctionBody()
Dinline_function_utils.cc58 static constexpr const char* const kDeviceRetOp = variable
59 FunctionLibraryDefinition::kDeviceRetOp;
Dfunction.cc61 static constexpr const char* const kDeviceRetOp = variable
62 FunctionLibraryDefinition::kDeviceRetOp;
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/
Dgraph-device-retval.pbtxt76 # ensure that kDeviceRetOp is used instead of kRetOp
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_ops.h173 Name(kDeviceRetOp).Device(DEVICE).TypeConstraint<int32>("T"), RetvalOp); \
/external/tensorflow/tensorflow/core/framework/
Dfunction.cc50 FunctionLibraryDefinition::kDeviceRetOp; member in tensorflow::FunctionLibraryDefinition
367 gnode->set_op(FunctionLibraryDefinition::kDeviceRetOp); in AddReturnNode()
618 n->op() == FunctionLibraryDefinition::kDeviceRetOp) { in Print()
Dfunction.h373 static constexpr const char* const kDeviceRetOp = "_DeviceRetval"; variable