Searched refs:feed_tensor (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | subgraph.cc | 219 Status ArgFeedRewrite::AddNode(Graph* g, NodeBuilder::NodeOut feed_tensor, in AddNode() argument
226 NodeBuilder(strings::StrCat("_arg_", feed_tensor.node->name(), "_", in AddNode()
227 feed_tensor.index, "_", arg_index_), in AddNode()
229 .Attr("T", BaseType(feed_tensor.node->output_type(feed_tensor.index))) in AddNode()
236 Status RecvFeedRewrite::AddNode(Graph* g, NodeBuilder::NodeOut feed_tensor, in AddNode() argument
239 NodeBuilder(strings::StrCat("_recv_", feed_tensor.node->name(), "_", in AddNode()
240 feed_tensor.index), in AddNode()
243 BaseType(feed_tensor.node->output_type(feed_tensor.index))) in AddNode()
|
| H A D | subgraph.h | 125 Status AddNode(Graph* g, NodeBuilder::NodeOut feed_tensor,
136 Status AddNode(Graph* g, NodeBuilder::NodeOut feed_tensor,
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | graph_execution_state.cc | 318 Status AddNode(Graph* g, NodeBuilder::NodeOut feed_tensor, in AddNode() argument
321 auto check_no_cycle_fn = [this, feed_tensor, &s](Node* n) { in AddNode()
322 if (n == feed_tensor.node) { in AddNode()
325 feed_tensor.node->name(), "\" and \"", from_tensor_.node->name(), in AddNode()
334 NodeBuilder(strings::StrCat("_identity_", feed_tensor.node->name(), "_", in AddNode()
335 feed_tensor.index), in AddNode()
343 feed_tensor.node->assigned_device_name()); in AddNode()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | master_test.cc | 103 const Tensor* feed_tensor = p.second; in RunStep() local
106 feed_tensor->AsProtoTensorContent(f->mutable_tensor()); in RunStep()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/ |
| H A D | training_arrays_v1.py | 229 for i, (input_data, feed_tensor) in enumerate(zip(ins, feed)):
230 if issparse(input_data) and not backend.is_sparse(feed_tensor):
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/client/ |
| H A D | session.py | 1448 feed_tensor = feed_map[feed_name][0]
1449 feed_dict[feed_tensor.ref()] = np_val
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_nodes_test.cc | 475 Tensor feed_tensor(DT_FLOAT, input_shape); in TEST() local
476 item.feed.push_back(std::make_pair("feed", feed_tensor)); in TEST()
|