Home
last modified time | relevance | path

Searched defs:arg_nodes (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dgraph_compiler_util.cc57 std::unordered_set<const Node*>* arg_nodes) { in AddArgNodes()
201 Status CollectArgNodes(const Graph& graph, std::vector<Node*>* arg_nodes) { in CollectArgNodes()
234 std::vector<Node*> arg_nodes; in CreateXlaArgs() local
H A Dtf2xla_util.cc173 absl::flat_hash_map<int, Node*> arg_nodes; in ReplaceArgUsageWithConstNode() local
201 absl::flat_hash_map<int, Node*> arg_nodes; in ReplaceRetvalInputWithArg() local
H A Drearrange_function_argument.cc163 const gtl::InlinedVector<Node*, 4>* arg_nodes, // non-absl ok in RearrangeArgNodes()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
H A Dimport_model.cc647 absl::InlinedVector<OutputTensor, 4> arg_nodes; in ConvertDeferredFunctions() local
1368 const FunctionBody& fbody, absl::InlinedVector<OutputTensor, 4>* arg_nodes, in GetArgsAndRetsFromFunctionBody()
1549 const absl::InlinedVector<OutputTensor, 4>& arg_nodes, in Convert()
1604 const absl::InlinedVector<OutputTensor, 4>& arg_nodes, in ConvertFunctionArgAndRets()
2428 absl::InlinedVector<OutputTensor, 4> arg_nodes; in Convert() local
2529 absl::InlinedVector<OutputTensor, 4>* arg_nodes, in InferMainFunctionType()
2662 mlir::MLIRContext* context, absl::InlinedVector<OutputTensor, 4>* arg_nodes, in GetArgsRetsAndTypesFromFunctionGraph()
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dprocess_function_library_runtime.cc393 const std::vector<Node*>& arg_nodes) { in SetArgShape()
461 const std::vector<Node*>& arg_nodes, const std::vector<Node*>& ret_nodes, in PinArgsAndRets()
722 std::vector<Node*>* arg_nodes, std::vector<Node*>* ret_nodes, in GetGraphAndArgRets()
799 std::vector<Node*> arg_nodes, ret_nodes; in InstantiateMultiDevice() local
H A Dfunction_body.h38 gtl::InlinedVector<Node*, 4> arg_nodes; member
H A Dpartitioning_utils.cc159 std::vector<std::pair<Node*, FunctionArgIndex>> arg_nodes; in UpdateArgAndRetvalMetadata() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/
H A Dtransform_graph_function.cc52 std::vector<Node*> arg_nodes, ret_nodes; in TransformGraphFunction() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dextract_outside_compilation_pass.cc89 Status GetArgDataTypes(const std::vector<Node*>& arg_nodes, in GetArgDataTypes()
136 std::vector<Node*> arg_nodes = GatherNodesWithType(*g, "_Arg"); in ReplaceArgNodesWithRecvAtHostNode() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/
H A Ddtensor_device.cc1172 std::vector<Node*> arg_nodes; in SelectGraphToExecute() local