Searched refs:kRetOp (Results 1 – 10 of 10) sorted by relevance
73 REGISTER_SYSTEM_KERNEL_BUILDER(Name(kRetOp).Device(DEVICE_CPU), RetvalOp);89 Name(kRetOp).Device(DEVICE_SYCL).TypeConstraint<type>("T"), RetvalOp);91 TF_CALL_bool(REGISTER) REGISTER_KERNEL_BUILDER(Name(kRetOp)130 Name(kRetOp).Device(DEVICE_GPU).TypeConstraint<type>("T"), RetvalOp);134 TF_CALL_bool(REGISTER) REGISTER_KERNEL_BUILDER(Name(kRetOp)142 REGISTER_KERNEL_BUILDER(Name(kRetOp)148 REGISTER_KERNEL_BUILDER(Name(kRetOp)
26 static const char* const kRetOp = FunctionLibraryDefinition::kRetOp; variable
221 Name(kRetOp).Device(DEVICE).TypeConstraint("T", TYPES), RetvalOp); \222 REGISTER_KERNEL_BUILDER(Name(kRetOp) \278 REGISTER_KERNEL_BUILDER(Name(FunctionLibraryDefinition::kRetOp) \
53 static constexpr const char* const kRetOp = FunctionLibraryDefinition::kRetOp; variable141 ndef.set_op(kRetOp); in AddRet()1930 if (n->type_string() == kRetOp || n->type_string() == kDeviceRetOp) { in FunctionBody()2034 DCHECK_EQ(y->type_string(), kRetOp); in Compute()
401 static constexpr const char* const kRetOp = "_Retval"; variable
343 gnode->set_op(FunctionLibraryDefinition::kRetOp); in AddReturnNode()590 } else if (n->op() == FunctionLibraryDefinition::kRetOp || in Print()
680 FunctionLibraryDefinition::kRetOp); in RegisterSegmentFunctionToFunctionLibrary()
581 retval_1.set_op(FunctionLibraryDefinition::kRetOp); in TEST_F()589 retval_2.set_op(FunctionLibraryDefinition::kRetOp); in TEST_F()
437 retval.set_op(FunctionLibraryDefinition::kRetOp); in TEST()
928 Status status = NodeBuilder(retval_name, FunctionLibraryDefinition::kRetOp) in CompileSingleOp()