Searched defs:arg_nodes (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | graph_compiler_util.cc | 57 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
|
| D | rearrange_function_argument.cc | 163 const gtl::InlinedVector<Node*, 4>* arg_nodes, // non-absl ok in RearrangeArgNodes()
|
| D | tf2xla_util.cc | 115 std::unordered_map<int, Node*> arg_nodes; in ReplaceArgUsageWithConstNode() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_body.h | 38 gtl::InlinedVector<Node*, 4> arg_nodes; member
|
| D | partitioning_utils.cc | 81 std::vector<std::pair<Node*, FunctionArgIndex>> arg_nodes; in UpdateArgAndRetvalMetadata() local
|
| D | process_function_library_runtime.cc | 370 const std::vector<Node*>& arg_nodes) { in SetArgShape()
425 const std::vector<Node*>& arg_nodes, const std::vector<Node*>& ret_nodes, in PinArgsAndRets()
635 std::vector<Node*>* arg_nodes, std::vector<Node*>* ret_nodes, in GetGraphAndArgRets()
712 std::vector<Node*> arg_nodes, ret_nodes; in InstantiateMultiDevice() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 1215 const FunctionBody& fbody, absl::InlinedVector<OutputTensor, 4>* arg_nodes, in GetArgsAndRetsFromFunctionBody()
1317 absl::InlinedVector<OutputTensor, 4> arg_nodes; in ConvertLibFunction() local
1425 const absl::InlinedVector<OutputTensor, 4>& arg_nodes, in Convert()
1480 const absl::InlinedVector<OutputTensor, 4>& arg_nodes, in ConvertFunctionArgAndRets()
2222 absl::InlinedVector<OutputTensor, 4> arg_nodes; in Convert() local
2315 absl::InlinedVector<OutputTensor, 4>* arg_nodes, in InferMainFunctionType()
2425 mlir::MLIRContext* context, absl::InlinedVector<OutputTensor, 4>* arg_nodes, in GetArgsRetsAndTypesFromFunctionGraph()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 88 Status GetArgDataTypes(const std::vector<Node*>& arg_nodes, in GetArgDataTypes()
137 std::vector<Node*> arg_nodes = GatherNodesWithType(*g, "_Arg"); in ReplaceArgNodesWithRecvAtHostNode() local
|