Searched refs:fetch_tensor (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | subgraph.cc | 256 Status RetvalFetchRewrite::AddNode(Graph* g, NodeBuilder::NodeOut fetch_tensor, in AddNode() argument
263 NodeBuilder(strings::StrCat("_retval_", fetch_tensor.node->name(), "_", in AddNode()
264 fetch_tensor.index, "_", retval_index_), in AddNode()
266 .Input(fetch_tensor.node, fetch_tensor.index) in AddNode()
268 BaseType(fetch_tensor.node->output_type(fetch_tensor.index))) in AddNode()
275 Status SendFetchRewrite::AddNode(Graph* g, NodeBuilder::NodeOut fetch_tensor, in AddNode() argument
278 NodeBuilder(strings::StrCat("_send_", fetch_tensor.node->name(), "_", in AddNode()
279 fetch_tensor.index), in AddNode()
281 .Input(fetch_tensor.node, fetch_tensor.index) in AddNode()
|
| H A D | subgraph.h | 146 Status AddNode(Graph* g, NodeBuilder::NodeOut fetch_tensor,
158 Status AddNode(Graph* g, NodeBuilder::NodeOut fetch_tensor,
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| H A D | grpc_session.cc | 458 Tensor fetch_tensor; in RunCallable() local
459 if (!fetch_tensor.FromProto(cpu_allocator(), fetch)) { in RunCallable()
463 fetch_tensors->push_back(std::move(fetch_tensor)); in RunCallable()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | function_optimizer.cc | 419 TensorId fetch_tensor = ParseTensorName(fetch); in GetActiveOutputs() local
420 if (fetch_tensor.node() == node.name()) { in GetActiveOutputs()
421 active_outputs.insert(fetch_tensor.index()); in GetActiveOutputs()
|