Searched refs:kDeviceRetOp (Results 1 – 8 of 8) sorted by relevance
27 static const char* const kDeviceRetOp = FunctionLibraryDefinition::kDeviceRetOp; variable
75 REGISTER_SYSTEM_KERNEL_BUILDER(Name(kDeviceRetOp).Device(DEVICE_CPU), RetvalOp);141 Name(kDeviceRetOp).Device(DEVICE_GPU).TypeConstraint<int32>("T"), RetvalOp);
76 # ensure that kDeviceRetOp is used instead of kRetOp
161 Name(kDeviceRetOp).Device(DEVICE).TypeConstraint<int32>("T"), RetvalOp); \
330 static constexpr const char* const kDeviceRetOp = "_DeviceRetval"; variable
50 FunctionLibraryDefinition::kDeviceRetOp; member in tensorflow::FunctionLibraryDefinition367 gnode->set_op(FunctionLibraryDefinition::kDeviceRetOp); in AddReturnNode()618 n->op() == FunctionLibraryDefinition::kDeviceRetOp) { in Print()
58 static constexpr const char* const kDeviceRetOp = variable59 FunctionLibraryDefinition::kDeviceRetOp;2355 if (n->type_string() == kRetOp || n->type_string() == kDeviceRetOp) { in FunctionBody()
1240 op == FunctionLibraryDefinition::kDeviceRetOp) { in ConvertFunctionArgAndRets()