Home
last modified time | relevance | path

Searched refs:input_nodes (Results 1 – 25 of 33) sorted by relevance

12

/external/tensorflow/tensorflow/contrib/framework/python/framework/
Dgraph_util.py37 def fuse_op(graph_def, input_nodes, output_nodes, output_dtypes, argument
59 if isinstance(input_nodes, six.string_types):
67 _assert_nodes_are_present(name_to_node, input_nodes + output_nodes)
70 reachable_by_input = _bfs_for_reachable_nodes(input_nodes, name_to_input_name)
76 input_nodes_set = set(input_nodes)
108 for node in input_nodes:
Dgraph_util_test.py30 def GetNewNode(name, op, input_nodes): argument
34 for node in input_nodes:
/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_constants_lib.cc55 StringPieceSet input_nodes; in ReplaceSendRecvs() local
62 input_nodes.insert(id.first); in ReplaceSendRecvs()
104 auto iter = input_nodes.find(new_node->name()); in ReplaceSendRecvs()
105 if (iter != input_nodes.end()) { in ReplaceSendRecvs()
106 input_nodes.erase(iter); in ReplaceSendRecvs()
112 for (StringPiece name : input_nodes) { in ReplaceSendRecvs()
Dstrip_unused_nodes.cc122 std::set<string> input_nodes; in StripUnusedNodes() local
125 input_nodes.insert(NodeNameFromInput(input)); in StripUnusedNodes()
143 if (input_nodes.count(current_input)) { in StripUnusedNodes()
171 if (input_nodes.count(node.name())) { in StripUnusedNodes()
Dbackports.cc36 [](const NodeMatch& match, const std::set<string>& input_nodes, in BackportConcatV2Transform()
73 const std::set<string>& input_nodes, in BackportTensorArrayV3Transform() argument
119 [](const NodeMatch& match, const std::set<string>& input_nodes, in BackportTensorArrayV3Transform() argument
Dfuse_convolutions.cc47 [](const NodeMatch& match, const std::set<string>& input_nodes, in FuseResizePadAndConv()
96 [](const NodeMatch& match, const std::set<string>& input_nodes, in FuseResizeAndConv() argument
156 [](const NodeMatch& match, const std::set<string>& input_nodes, in FusePadAndConv() argument
Dquantize_nodes.cc281 const std::set<string>& input_nodes, in RemoveRedundantQuantizations()
404 [](const NodeMatch& match, const std::set<string>& input_nodes, in ConvertFakeQuantsToRequantize()
498 [](const NodeMatch& match, const std::set<string>& input_nodes, in MergeAdjacentRequantizes()
548 [depth](const NodeMatch& match, const std::set<string>& input_nodes, in HoistFakeQuants()
697 const NodeMatch& match, const std::set<string>& input_nodes, in QuantizeNodes()
Dremove_ema.cc111 [](const NodeMatch& match, const std::set<string>& input_nodes, in RemoveEMA()
Dfold_old_batch_norms.cc276 const std::set<string>& input_nodes, in FoldOldBatchNorms()
319 const std::set<string>& input_nodes, in FoldOldBatchNorms()
Dround_weights.cc41 [num_steps](const NodeMatch& match, const std::set<string>& input_nodes, in RoundWeights()
Dfold_batch_norms.cc50 [](const NodeMatch& match, const std::set<string>& input_nodes, in FoldBatchNorms()
Dremove_nodes.cc72 const NodeMatch& match, const std::set<string>& input_nodes, in RemoveNodes()
Dflatten_atrous.cc51 [](const NodeMatch& match, const std::set<string>& input_nodes, in FlattenAtrousConv()
Dquantize_weights.cc44 const std::set<string>& input_nodes, in QuantizeWeights()
Dtransform_utils.cc395 std::set<string> input_nodes; in ReplaceMatchingOpTypes() local
404 input_nodes.insert(matched_node.name()); in ReplaceMatchingOpTypes()
424 node_generator(*match, input_nodes, output_nodes, &new_nodes)); in ReplaceMatchingOpTypes()
/external/tensorflow/tensorflow/c/
Dc_api_function.cc390 std::unordered_map<const Node*, std::vector<int>>* input_nodes) in ProcessInputs() argument
407 const auto& iter = input_nodes->find(&node); in ProcessInputs()
408 if (iter == input_nodes->end()) { in ProcessInputs()
409 input_nodes->insert({&node, {idx}}); in ProcessInputs()
449 const std::unordered_map<const Node*, std::vector<int>>& input_nodes, in ComputeBodyNodes() argument
454 const auto& iter = input_nodes.find(node); in ComputeBodyNodes()
455 if (iter == input_nodes.end()) { in ComputeBodyNodes()
499 std::unordered_map<const Node*, std::vector<int>> input_nodes; in TF_GraphToFunction() local
501 &input_tensors, &input_nodes); in TF_GraphToFunction()
522 fn_body, fn_name, num_opers, opers, input_nodes, &body_nodes); in TF_GraphToFunction()
/external/autotest/server/site_tests/audio_AudioBasicExternalMicrophone/
Daudio_AudioBasicExternalMicrophone.py64 _, input_nodes = audio_facade.get_selected_node_types()
65 if input_nodes != ['MIC']:
68 'device' % input_nodes)
/external/autotest/client/common_lib/cros/cfm/
Dcras_node_collector_unittest.py67 input_nodes = (
85 mock_host.run_output.side_effect = [input_devices, input_nodes]
/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer.cc286 std::vector<NodeDef*> input_nodes; in OptimizeNode() local
290 input_nodes.push_back(input_node); in OptimizeNode()
296 input_nodes[0]->device() != node->device()) || in OptimizeNode()
303 const string& in0_dev = input_nodes[0]->device(); in OptimizeNode()
304 const string& in1_dev = input_nodes[1]->device(); in OptimizeNode()
335 const NodeDef* input = input_nodes[i]; in OptimizeNode()
Dauto_parallel.cc161 std::vector<const NodeDef*> input_nodes; in Initialize() local
164 input_nodes = ComputeTransitiveFanin(graph_, {dequeue_node->name()}); in Initialize()
166 LOG(INFO) << "Number of input nodes: " << input_nodes.size(); in Initialize()
178 for (const auto& input_node : input_nodes) { in Initialize()
/external/tensorflow/tensorflow/contrib/tensorrt/shape_fn/
Dtrt_shfn.cc59 std::vector<string> input_nodes; in TRTEngineOpShapeInference() local
60 TF_RETURN_IF_ERROR(context->GetAttr("input_nodes", &input_nodes)); in TRTEngineOpShapeInference()
/external/autotest/client/cros/audio/
Dcras_utils.py207 input_nodes = []
212 input_nodes.append(node['Id'])
215 return (output_nodes, input_nodes)
/external/tensorflow/tensorflow/python/framework/
Dgraph_util_impl.py303 input_nodes = input_graph.node
305 for node in input_nodes:
310 for node in input_nodes:
/external/tensorflow/tensorflow/contrib/meta_graph_transform/
Dmeta_graph_transform_test.py369 input_nodes = 'input_nodes'
416 base_meta_graph_def, [input_nodes], [output_nodes],
424 transformed_graph_def, [input_nodes], [
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transferer.cc323 std::vector<string> input_nodes; in TransformGraphToAddAggregatedInputNode() local
327 input_nodes.emplace_back(node->name()); in TransformGraphToAddAggregatedInputNode()
364 std::vector<Node*> original_input_nodes(input_nodes.size()); in TransformGraphToAddAggregatedInputNode()
366 for (int i = 0; i < input_nodes.size(); ++i) { in TransformGraphToAddAggregatedInputNode()
367 const string& node_name = input_nodes.at(i); in TransformGraphToAddAggregatedInputNode()

12