Searched defs:ret_nodes (Results 1 – 6 of 6) 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 | 82 std::vector<std::pair<Node*, int>> ret_nodes; in UpdateArgAndRetvalMetadata() local
|
| D | process_function_library_runtime.cc | 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/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()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 1216 absl::InlinedVector<OutputTensor, 4>* ret_nodes, in GetArgsAndRetsFromFunctionBody()
1318 absl::InlinedVector<OutputTensor, 4> ret_nodes; in ConvertLibFunction() local
1426 const absl::InlinedVector<OutputTensor, 4>& ret_nodes, in Convert()
1481 const absl::InlinedVector<OutputTensor, 4>& ret_nodes, in ConvertFunctionArgAndRets()
2223 absl::InlinedVector<OutputTensor, 4> ret_nodes; in Convert() local
2316 absl::InlinedVector<OutputTensor, 4>* ret_nodes) { in InferMainFunctionType()
2426 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()
254 std::vector<Node*> ret_nodes = GatherNodesWithType(*g, "_Retval"); in ReplaceRetNodesWithSendFromHostNode() local
|