Home
last modified time | relevance | path

Searched refs:IsRetval (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dgraph_to_functiondef.cc332 if (node->IsRetval()) { in GraphToFunctionDefHelper()
414 if (node->IsRetval()) { in GraphToFunctionDef()
508 if (outputs[r].node->IsRetval()) { in GraphToFunctionDef()
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_functional_ops.cc77 if (n->IsArg() || n->IsRetval()) return true; in HasArgsOrRetvals()
Dpartitioning_utils.cc91 } else if (node->IsRetval()) { in UpdateArgAndRetvalMetadata()
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.cc370 if (IsRetval(node)) { in RemoveFunctionOutputs()
574 if (IsArg(func_node) || IsRetval(func_node) || in MakeFunctionDef()
/external/tensorflow/tensorflow/core/grappler/
Dop_types.h147 bool IsRetval(const NodeDef& node);
Dop_types.cc449 bool IsRetval(const NodeDef& node) { in IsRetval() function
/external/tensorflow/tensorflow/core/graph/
Dgraph.h194 bool IsRetval() const { return class_ == NC_RETVAL; } in IsRetval() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer.cc80 if (node.input_size() > 1 && (IsRetval(*consumer) || IsMerge(*consumer))) { in SafeToRemoveIdentity()
Dfunction_optimizer.cc865 return n->op_def().is_stateful() && !n->IsArg() && !n->IsRetval() && in ValidateSideEffectsExecution()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.cc102 } else if (n->IsRetval()) { in ComputeArgAndRetvalShardings()
740 if (n->IsRetval()) { in CompileFunction()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc2034 else if (node->IsRetval()) in GetArgsRetsAndTypesFromFunctionGraph()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc5755 } else if (tensorflow::grappler::IsRetval(node_def)) { in ConvertGraphDefToEngine()