Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/framework/
Dgraph_to_functiondef.cc326 if (node->IsRetval()) { in GraphToFunctionDefHelper()
408 if (node->IsRetval()) { in GraphToFunctionDef()
512 if (outputs[r].node->IsRetval()) { in GraphToFunctionDef()
/external/tensorflow/tensorflow/core/common_runtime/
Dlower_functional_ops.cc73 if (n->IsArg() || n->IsRetval()) return true; in HasArgsOrRetvals()
Dpartitioning_utils.cc96 } else if (node->IsRetval()) { in UpdateArgAndRetvalMetadata()
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.cc385 if (IsRetval(node)) { in RemoveFunctionOutputs()
597 if (IsArg(func_node) || IsRetval(func_node) || in MakeFunctionDef()
/external/tensorflow/tensorflow/core/grappler/
Dop_types.h157 bool IsRetval(const NodeDef& node);
Dop_types.cc491 bool IsRetval(const NodeDef& node) { in IsRetval() function
/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
Dencapsulate_tpu_computations_pass.cc1014 if (!e->dst()->IsRetval() && in MoveTailOutsideCompilationToHost()
1133 if (e->dst()->IsRetval()) { in MoveTailOutsideCompilationToHost()
1211 if (n->IsRetval() && n->in_edges().empty()) { in MoveTailOutsideCompilationToHost()
1382 if (!n->IsRetval()) { in RemoveUnusedXlaOutput()
1498 if (!e->IsControlEdge() && e->src()->IsArg() && e->dst()->IsRetval()) { in RemoveEdgesBetweenArgAndRetval()
1832 if (!e->dst()->IsRetval() && in FindArgsToLiftForWhileNode()
Ddistributed_tpu_rewrite_pass.cc1932 } else if (node->IsRetval()) { in AssignArgsAndRetvalsToCores()
/external/tensorflow/tensorflow/core/graph/
Dgraph.h203 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.cc881 return n->op_def().is_stateful() && !n->IsArg() && !n->IsRetval() && in ValidateSideEffectsExecution()
Dconstant_folding.cc290 if (IsRetval(*consumer)) { in ForwardInputs()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.cc110 } else if (n->IsRetval()) { in ComputeArgAndRetvalShardings()
795 if (n->IsRetval()) { in CompileFunction()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc1555 if (ret.node->IsRetval()) { in ConvertFunctionArgAndRets()
2452 else if (node->IsRetval()) in GetArgsRetsAndTypesFromFunctionGraph()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc6397 } else if (tensorflow::grappler::IsRetval(node_def)) { in ConvertGraphDefToEngine()