Searched defs:ret_nodes (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_body.h | 41 gtl::InlinedVector<Node*, 4> ret_nodes; member
|
| D | partitioning_utils.cc | 160 std::vector<std::pair<Node*, int>> ret_nodes; in UpdateArgAndRetvalMetadata() local
|
| D | process_function_library_runtime.cc | 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/tf2xla/ |
| D | rearrange_function_argument.cc | 180 const gtl::InlinedVector<Node*, 4>& ret_nodes, // non-absl ok in CalculateRetvalRearrange()
262 const gtl::InlinedVector<Node*, 4>& ret_nodes, // non-absl ok in RearrangeRetvalNodes()
|
| D | tf2xla_util.cc | 202 absl::flat_hash_map<int, Node*> ret_nodes; in ReplaceRetvalInputWithArg() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 648 absl::InlinedVector<OutputTensor, 4> ret_nodes; in ConvertDeferredFunctions() local
1369 absl::InlinedVector<OutputTensor, 4>* ret_nodes, in GetArgsAndRetsFromFunctionBody()
1550 const absl::InlinedVector<OutputTensor, 4>& ret_nodes, in Convert()
1605 const absl::InlinedVector<OutputTensor, 4>& ret_nodes, in ConvertFunctionArgAndRets()
2429 absl::InlinedVector<OutputTensor, 4> ret_nodes; in Convert() local
2530 absl::InlinedVector<OutputTensor, 4>* ret_nodes) { in InferMainFunctionType()
2663 absl::InlinedVector<OutputTensor, 4>* ret_nodes) { in GetArgsRetsAndTypesFromFunctionGraph()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | extract_outside_compilation_pass.cc | 189 Status GetRetDataTypes(const std::vector<Node*>& ret_nodes, in GetRetDataTypes()
210 const std::vector<Node*>& ret_nodes, in BuildSendFromHostNode()
253 std::vector<Node*> ret_nodes = GatherNodesWithType(*g, "_Retval"); in ReplaceRetNodesWithSendFromHostNode() local
|