| /external/tensorflow/tensorflow/contrib/tensorrt/shape_fn/ |
| trt_shfn.cc | 59 std::vector<string> input_nodes; local
60 TF_RETURN_IF_ERROR(context->GetAttr("input_nodes", &input_nodes));
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| strip_unused_nodes.cc | 122 std::set<string> input_nodes; local
125 input_nodes.insert(NodeNameFromInput(input));
143 if (input_nodes.count(current_input)) {
171 if (input_nodes.count(node.name())) {
|
| fold_constants_lib.cc | 55 StringPieceSet input_nodes; local
62 input_nodes.insert(id.first);
102 // node name from input_nodes so that a duplicate does not get added to the
104 auto iter = input_nodes.find(new_node->name());
105 if (iter != input_nodes.end()) {
106 input_nodes.erase(iter);
112 for (StringPiece name : input_nodes) {
|
| transform_utils.cc | 395 std::set<string> input_nodes; local
404 input_nodes.insert(matched_node.name());
424 node_generator(*match, input_nodes, output_nodes, &new_nodes));
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| auto_parallel.cc | 161 std::vector<const NodeDef*> input_nodes; local
164 input_nodes = ComputeTransitiveFanin(graph_, {dequeue_node->name()});
166 LOG(INFO) << "Number of input nodes: " << input_nodes.size();
178 for (const auto& input_node : input_nodes) {
|
| dependency_optimizer.cc | 286 std::vector<NodeDef*> input_nodes; local
290 input_nodes.push_back(input_node);
296 input_nodes[0]->device() != node->device()) ||
303 const string& in0_dev = input_nodes[0]->device();
304 const string& in1_dev = input_nodes[1]->device();
335 const NodeDef* input = input_nodes[i];
|
| /external/tensorflow/tensorflow/c/ |
| c_api_function.cc | 383 // Converts `ninputs` and `inputs` into `inputs_tensors` and `input_nodes` and
384 // does various checks while doing so. `input_nodes` will contain the same
390 std::unordered_map<const Node*, std::vector<int>>* input_nodes)
407 const auto& iter = input_nodes->find(&node);
408 if (iter == input_nodes->end()) {
409 input_nodes->insert({&node, {idx}});
449 const std::unordered_map<const Node*, std::vector<int>>& input_nodes,
454 const auto& iter = input_nodes.find(node);
455 if (iter == input_nodes.end()) {
499 std::unordered_map<const Node*, std::vector<int>> input_nodes; local
[all...] |
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| graph_transferer.cc | 323 std::vector<string> input_nodes; local
327 input_nodes.emplace_back(node->name());
364 std::vector<Node*> original_input_nodes(input_nodes.size());
366 for (int i = 0; i < input_nodes.size(); ++i) {
367 const string& node_name = input_nodes.at(i);
[all...] |
| /external/adhd/cras/src/common/ |
| cras_types.h | 293 * input_nodes - Input nodes currently attached.
330 struct cras_ionode_info input_nodes[CRAS_MAX_IONODES]; member in struct:cras_server_state
|
