Home
last modified time | relevance | path

Searched refs:kRetOp (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dfunction_ops.h26 static const char* const kRetOp = FunctionLibraryDefinition::kRetOp; variable
Dfunction_ops.cc89 REGISTER_SYSTEM_KERNEL_BUILDER(Name(kRetOp).Device(DEVICE_CPU), RetvalOp);
95 REGISTER_KERNEL_BUILDER(Name(kRetOp).Device(DEVICE_TPU_SYSTEM), RetvalOp);
129 Name(kRetOp).Device(DEVICE_GPU).TypeConstraint<type>("T"), RetvalOp);
133 TF_CALL_bool(REGISTER) REGISTER_KERNEL_BUILDER(Name(kRetOp)
141 REGISTER_KERNEL_BUILDER(Name(kRetOp)
147 REGISTER_KERNEL_BUILDER(Name(kRetOp)
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_ops.h166 Name(kRetOp).Device(DEVICE).TypeConstraint("T", TYPES), RetvalOp); \
167 REGISTER_KERNEL_BUILDER(Name(kRetOp) \
227 REGISTER_KERNEL_BUILDER(Name(FunctionLibraryDefinition::kRetOp) \
Dxla_compilation_cache.cc251 Status status = NodeBuilder(retval_name, FunctionLibraryDefinition::kRetOp) in CreateGraph()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dgraph_compiler_util.cc133 FunctionLibraryDefinition::kRetOp) in AddRetvalNodes()
180 } else if (n->type_string() == FunctionLibraryDefinition::kRetOp) { in RewriteAndPruneGraph()
184 return errors::Aborted(FunctionLibraryDefinition::kRetOp, in RewriteAndPruneGraph()
Dxla_compiler_test.cc657 retval_1.set_op(FunctionLibraryDefinition::kRetOp); in TEST_F()
665 retval_2.set_op(FunctionLibraryDefinition::kRetOp); in TEST_F()
Dfunctionalize_control_flow_test.cc543 retval.set_op(FunctionLibraryDefinition::kRetOp); in TEST()
/external/tensorflow/tensorflow/core/common_runtime/
Dfunction_body.cc34 if (n->type_string() == FunctionLibraryDefinition::kRetOp || in FunctionBody()
Dinline_function_utils.cc57 static constexpr const char* const kRetOp = FunctionLibraryDefinition::kRetOp; variable
Dfunction.cc60 static constexpr const char* const kRetOp = FunctionLibraryDefinition::kRetOp; variable
120 ndef.set_op(kRetOp); in AddRet()
1459 DCHECK_EQ(y->type_string(), kRetOp); in Compute()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/graphdef2mlir/
Dgraph-device-retval.pbtxt76 # ensure that kDeviceRetOp is used instead of kRetOp
/external/tensorflow/tensorflow/core/kernels/data/
Dserialization_utils.cc35 if (node.op() == FunctionLibraryDefinition::kRetOp) continue; in FindStatefulOps()
Ddataset_ops.cc129 if (node.op() == FunctionLibraryDefinition::kRetOp) { in Compute()
/external/tensorflow/tensorflow/core/framework/
Dfunction.cc48 /* static */ constexpr const char* const FunctionLibraryDefinition::kRetOp; member in tensorflow::FunctionLibraryDefinition
369 gnode->set_op(FunctionLibraryDefinition::kRetOp); in AddReturnNode()
617 } else if (n->op() == FunctionLibraryDefinition::kRetOp || in Print()
Dfunction.h372 static constexpr const char* const kRetOp = "_Retval"; variable
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dexport_graphdef.cc262 node_def->set_op(FunctionLibraryDefinition::kRetOp); in GetReturnNode()
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc1186 FunctionLibraryDefinition::kRetOp); in MoveTailOutsideCompilationToHost()