Searched defs:arg_nodes (Results 1 – 10 of 10) 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 | tf2xla_util.cc | 173 absl::flat_hash_map<int, Node*> arg_nodes; in ReplaceArgUsageWithConstNode() local
201 absl::flat_hash_map<int, Node*> arg_nodes; in ReplaceRetvalInputWithArg() local
|
| D | rearrange_function_argument.cc | 163 const gtl::InlinedVector<Node*, 4>* arg_nodes, // non-absl ok in RearrangeArgNodes()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_body.h | 38 gtl::InlinedVector<Node*, 4> arg_nodes; member
|
| D | partitioning_utils.cc | 159 std::vector<std::pair<Node*, FunctionArgIndex>> arg_nodes; in UpdateArgAndRetvalMetadata() local
|
| D | process_function_library_runtime.cc | 393 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
|
| /external/tensorflow/tensorflow/core/tfrt/eager/ |
| D | transform_graph_function.cc | 52 std::vector<Node*> arg_nodes, ret_nodes; in TransformGraphFunction() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 647 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()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 89 Status GetArgDataTypes(const std::vector<Node*>& arg_nodes, in GetArgDataTypes()
136 std::vector<Node*> arg_nodes = GatherNodesWithType(*g, "_Arg"); in ReplaceArgNodesWithRecvAtHostNode() local
|
| /external/tensorflow/tensorflow/dtensor/cc/ |
| D | dtensor_device.cc | 1172 std::vector<Node*> arg_nodes; in SelectGraphToExecute() local
|