Searched refs:feed_tensor (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/graph/ |
| 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()
|
| D | subgraph.h | 125 Status AddNode(Graph* g, NodeBuilder::NodeOut feed_tensor,
136 Status AddNode(Graph* g, NodeBuilder::NodeOut feed_tensor,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_execution_state.cc | 317 Status AddNode(Graph* g, NodeBuilder::NodeOut feed_tensor, in AddNode() argument
320 auto check_no_cycle_fn = [this, feed_tensor, &s](Node* n) { in AddNode()
321 if (n == feed_tensor.node) { in AddNode()
324 feed_tensor.node->name(), "\" and \"", from_tensor_.node->name(), in AddNode()
333 NodeBuilder(strings::StrCat("_identity_", feed_tensor.node->name(), "_", in AddNode()
334 feed_tensor.index), in AddNode()
342 feed_tensor.node->assigned_device_name()); in AddNode()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | master_test.cc | 103 const Tensor* feed_tensor = p.second; in RunStep() local
106 feed_tensor->AsProtoTensorContent(f->mutable_tensor()); in RunStep()
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_arrays_v1.py | 233 for i, (input_data, feed_tensor) in enumerate(zip(ins, feed)):
234 if issparse(input_data) and not K.is_sparse(feed_tensor):
|
| /external/tensorflow/tensorflow/python/client/ |
| D | session.py | 1445 feed_tensor = feed_map[feed_name][0]
1446 feed_dict[feed_tensor.ref()] = np_val
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 701 Tensor feed_tensor(DT_FLOAT, input_shape); in TEST() local
702 item.feed.push_back(std::make_pair("feed", feed_tensor)); in TEST()
|