HomeSort by relevance Sort by last modified time
    Searched refs:kRetOp (Results 1 - 10 of 10) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
function_ops.h 26 static const char* const kRetOp = FunctionLibraryDefinition::kRetOp;
function_ops.cc 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)
  /external/tensorflow/tensorflow/compiler/jit/
xla_device_ops.h 221 Name(kRetOp).Device(DEVICE).TypeConstraint("T", TYPES), RetvalOp); \
222 REGISTER_KERNEL_BUILDER(Name(kRetOp) \
278 REGISTER_KERNEL_BUILDER(Name(FunctionLibraryDefinition::kRetOp) \
  /external/tensorflow/tensorflow/core/common_runtime/
function.cc 53 static constexpr const char* const kRetOp = FunctionLibraryDefinition::kRetOp;
141 ndef.set_op(kRetOp);
    [all...]
  /external/tensorflow/tensorflow/core/framework/
function.h 398 // created for return values bear the name given by `kRetOp`.
401 static constexpr const char* const kRetOp = "_Retval";
    [all...]
function.cc 343 gnode->set_op(FunctionLibraryDefinition::kRetOp);
590 } else if (n->op() == FunctionLibraryDefinition::kRetOp ||
    [all...]
  /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
convert_graph.cc 680 FunctionLibraryDefinition::kRetOp);
    [all...]
  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_compiler_test.cc 581 retval_1.set_op(FunctionLibraryDefinition::kRetOp);
589 retval_2.set_op(FunctionLibraryDefinition::kRetOp);
    [all...]
functionalize_control_flow_test.cc 437 retval.set_op(FunctionLibraryDefinition::kRetOp);
    [all...]
xla_compiler.cc     [all...]

Completed in 1703 milliseconds