Home
last modified time | relevance | path

Searched refs:input_nodes (Results 1 – 25 of 43) 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)
113 for node in input_nodes:
Dgraph_util_test.py30 def GetNewNode(name, op, input_nodes): argument
34 for node in input_nodes:
/external/libaom/libaom/av1/encoder/x86/
Dml_sse3.c153 void av1_nn_predict_sse3(const float *input_nodes, in av1_nn_predict_sse3() argument
174 nn_propagate_4to8(&input_nodes[in], in av1_nn_predict_sse3()
186 nn_propagate_8to4(&input_nodes[in], in av1_nn_predict_sse3()
197 nn_propagate_4to4(&input_nodes[in], in av1_nn_predict_sse3()
208 nn_propagate_8to1(&input_nodes[in], in av1_nn_predict_sse3()
218 nn_propagate_4to1(&input_nodes[in], in av1_nn_predict_sse3()
230 __m128 input = _mm_load1_ps(&input_nodes[in_node]); in av1_nn_predict_sse3()
239 input_nodes = output_nodes; in av1_nn_predict_sse3()
/external/libaom/libaom/av1/encoder/
Dml.c21 void av1_nn_predict_c(const float *input_nodes, in av1_nn_predict_c() argument
39 val += layer_weights[node * num_input_nodes + i] * input_nodes[i]; in av1_nn_predict_c()
45 input_nodes = output_nodes; in av1_nn_predict_c()
55 val += layer_weights[node * num_input_nodes + i] * input_nodes[i]; in av1_nn_predict_c()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dstrip_unused_nodes.cc121 std::set<string> input_nodes; in StripUnusedNodes() local
124 input_nodes.insert(NodeNameFromInput(input)); in StripUnusedNodes()
142 if (input_nodes.count(current_input)) { in StripUnusedNodes()
170 if (input_nodes.count(node.name())) { in StripUnusedNodes()
Dfold_constants_lib.cc54 StringPieceSet input_nodes; in ReplaceSendRecvs() local
61 input_nodes.insert(id.first); in ReplaceSendRecvs()
103 auto iter = input_nodes.find(new_node->name()); in ReplaceSendRecvs()
104 if (iter != input_nodes.end()) { in ReplaceSendRecvs()
105 input_nodes.erase(iter); in ReplaceSendRecvs()
111 for (StringPiece name : input_nodes) { in ReplaceSendRecvs()
Dbackports.cc35 [](const NodeMatch& match, const std::set<string>& input_nodes, in BackportConcatV2Transform()
72 const std::set<string>& input_nodes, in BackportTensorArrayV3Transform() argument
118 [](const NodeMatch& match, const std::set<string>& input_nodes, in BackportTensorArrayV3Transform() argument
Dfuse_convolutions.cc46 [](const NodeMatch& match, const std::set<string>& input_nodes, in FuseResizePadAndConv()
95 [](const NodeMatch& match, const std::set<string>& input_nodes, in FuseResizeAndConv() argument
155 [](const NodeMatch& match, const std::set<string>& input_nodes, in FusePadAndConv() argument
Dfold_old_batch_norms.cc312 const std::set<string>& input_nodes, in FoldOldBatchNorms()
349 const std::set<string>& input_nodes, in FoldOldBatchNorms()
392 const std::set<string>& input_nodes, in FoldOldBatchNorms()
Dquantize_nodes.cc280 const std::set<string>& input_nodes, in RemoveRedundantQuantizations()
403 [](const NodeMatch& match, const std::set<string>& input_nodes, in ConvertFakeQuantsToRequantize()
497 [](const NodeMatch& match, const std::set<string>& input_nodes, in MergeAdjacentRequantizes()
547 [depth](const NodeMatch& match, const std::set<string>& input_nodes, in HoistFakeQuants()
696 const NodeMatch& match, const std::set<string>& input_nodes, in QuantizeNodes()
Dround_weights.cc40 [num_steps](const NodeMatch& match, const std::set<string>& input_nodes, in RoundWeights()
Dremove_nodes.cc71 const NodeMatch& match, const std::set<string>& input_nodes, in RemoveNodes()
Dfold_batch_norms.cc49 [](const NodeMatch& match, const std::set<string>& input_nodes, in FoldBatchNorms()
Dflatten_atrous.cc50 [](const NodeMatch& match, const std::set<string>& input_nodes, in FlattenAtrousConv()
/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/tensorflow/tensorflow/c/
Dc_api_function.cc463 std::unordered_map<const Node*, std::vector<int>>* input_nodes) in ProcessInputs() argument
480 const auto& iter = input_nodes->find(node); in ProcessInputs()
481 if (iter == input_nodes->end()) { in ProcessInputs()
482 input_nodes->insert({node, {idx}}); in ProcessInputs()
522 const std::unordered_map<const Node*, std::vector<int>>& input_nodes, in ComputeBodyNodes() argument
527 const auto& iter = input_nodes.find(node); in ComputeBodyNodes()
528 if (iter == input_nodes.end()) { in ComputeBodyNodes()
572 std::unordered_map<const Node*, std::vector<int>> input_nodes; in TF_GraphToFunctionWithControlOutputs() local
574 &input_tensors, &input_nodes); in TF_GraphToFunctionWithControlOutputs()
604 fn_body, fn_name, num_opers, opers, input_nodes, &body_nodes); in TF_GraphToFunctionWithControlOutputs()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer.cc166 const NodeDef& node, const std::vector<NodeDef*>& input_nodes, in BypassingNodeIsBeneficial() argument
181 input_nodes[0]->device() != node.device()) || in BypassingNodeIsBeneficial()
192 for (NodeDef* input_node : input_nodes) { in BypassingNodeIsBeneficial()
203 for (NodeDef* input_node : input_nodes) { in BypassingNodeIsBeneficial()
357 std::vector<NodeDef*> input_nodes; in OptimizeNode() local
364 input_nodes.push_back(input_node); in OptimizeNode()
367 if (!BypassingNodeIsBeneficial(*node, input_nodes, output_nodes)) { in OptimizeNode()
377 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()
Ddependency_optimizer.h48 const NodeDef& node, const std::vector<NodeDef*>& 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/autotest/client/cros/audio/
Dcras_utils.py234 input_nodes = []
239 input_nodes.append(node['Id'])
242 return (output_nodes, input_nodes)
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dchoose_fastest_dataset_op.cc164 std::vector<Node*> input_nodes; in AsGraphDefInternal() local
165 input_nodes.reserve(inputs_.size()); in AsGraphDefInternal()
169 input_nodes.push_back(input_node); in AsGraphDefInternal()
174 this, {}, {std::make_pair(0, input_nodes)}, in AsGraphDefInternal()
/external/tensorflow/tensorflow/python/framework/
Dgraph_util_impl.py335 input_nodes = input_graph.node
337 for node in input_nodes:
342 for node in input_nodes:
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc410 std::vector<Node*> input_nodes; in CreateTRTNode() local
487 input_nodes.push_back(CHECK_NOTNULL(input_node)); in CreateTRTNode()
593 VLOG(1) << "input_nodes size = " << input_nodes.size(); in CreateTRTNode()
594 for (int i = 0; i < input_nodes.size(); ++i) { in CreateTRTNode()
595 Node* n = CHECK_NOTNULL(input_nodes[i]); in CreateTRTNode()
/external/tensorflow/tensorflow/lite/python/
Dcreate_custom_op.py85 input_nodes=inputs,

12