Searched refs:fetch_tensor (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| D | subgraph.cc | 257 Status RetvalFetchRewrite::AddNode(Graph* g, NodeBuilder::NodeOut fetch_tensor, in AddNode() argument
264 NodeBuilder(strings::StrCat("_retval_", fetch_tensor.node->name(), "_", in AddNode()
265 fetch_tensor.index, "_", retval_index_), in AddNode()
267 .Input(fetch_tensor.node, fetch_tensor.index) in AddNode()
269 BaseType(fetch_tensor.node->output_type(fetch_tensor.index))) in AddNode()
276 Status SendFetchRewrite::AddNode(Graph* g, NodeBuilder::NodeOut fetch_tensor, in AddNode() argument
279 NodeBuilder(strings::StrCat("_send_", fetch_tensor.node->name(), "_", in AddNode()
280 fetch_tensor.index), in AddNode()
282 .Input(fetch_tensor.node, fetch_tensor.index) in AddNode()
|
| D | subgraph.h | 146 Status AddNode(Graph* g, NodeBuilder::NodeOut fetch_tensor,
158 Status AddNode(Graph* g, NodeBuilder::NodeOut fetch_tensor,
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_session.cc | 439 Tensor fetch_tensor; in RunCallable() local
440 if (!fetch_tensor.FromProto(cpu_allocator(), fetch)) { in RunCallable()
444 fetch_tensors->push_back(std::move(fetch_tensor)); in RunCallable()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | function_optimizer.cc | 504 TensorId fetch_tensor = ParseTensorName(fetch); in GetActiveOutputs() local
505 if (fetch_tensor.node() == node.name()) { in GetActiveOutputs()
506 active_outputs.insert(fetch_tensor.index()); in GetActiveOutputs()
|