Home
last modified time | relevance | path

Searched refs:output_nodes (Results 1 – 25 of 46) 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
62 if isinstance(output_nodes, six.string_types):
67 _assert_nodes_are_present(name_to_node, input_nodes + output_nodes)
72 reachable_by_output = _bfs_for_reachable_nodes(output_nodes,
79 output_nodes_set = set(output_nodes)
122 for index, n in enumerate(output_nodes):
/external/libaom/libaom/av1/encoder/x86/
Dml_sse3.c165 float *const output_nodes = output_layer ? output : buf[buf_index]; in av1_nn_predict_sse3() local
179 _mm_storeu_ps(&output_nodes[out + 4], out_h); in av1_nn_predict_sse3()
180 _mm_storeu_ps(&output_nodes[out], out_l); in av1_nn_predict_sse3()
191 _mm_storeu_ps(&output_nodes[out], outputs); in av1_nn_predict_sse3()
202 _mm_storeu_ps(&output_nodes[out], outputs); in av1_nn_predict_sse3()
212 output_nodes[out] = _mm_cvtss_f32(total); in av1_nn_predict_sse3()
222 output_nodes[out] = _mm_cvtss_f32(total); in av1_nn_predict_sse3()
236 output_nodes[out] = _mm_cvtss_f32(total); in av1_nn_predict_sse3()
239 input_nodes = output_nodes; in av1_nn_predict_sse3()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer.cc129 const NodeDef& node, const std::vector<NodeDef*>& output_nodes) const { in NumEdgesIfBypassed()
132 const int num_outputs = output_nodes.size(); in NumEdgesIfBypassed()
147 for (auto consumer : output_nodes) { in NumEdgesIfBypassed()
167 const std::vector<NodeDef*>& output_nodes) const { in BypassingNodeIsBeneficial()
171 const int num_outputs = output_nodes.size(); in BypassingNodeIsBeneficial()
174 if (NumEdgesIfBypassed(node, output_nodes) > num_inputs + num_outputs) { in BypassingNodeIsBeneficial()
183 output_nodes[0]->device() != node.device())) { in BypassingNodeIsBeneficial()
196 for (NodeDef* output_node : output_nodes) { in BypassingNodeIsBeneficial()
204 for (NodeDef* output_node : output_nodes) { in BypassingNodeIsBeneficial()
236 const std::set<NodeDef*> output_nodes = node_map_->GetOutputs(node_name); in OptimizeNode() local
[all …]
Ddependency_optimizer.h49 const std::vector<NodeDef*>& output_nodes) const;
51 const std::vector<NodeDef*>& output_nodes) const;
Dfunction_optimizer.cc825 num_func_outputs += out_arg.output_nodes.size(); in SpecializeFunction()
1000 for (const string& output_node : output_arg.output_nodes) { in InlinedFunctionOutputsNode()
1041 absl::flat_hash_set<absl::string_view> output_nodes; in InlineDirectFunctionCall() local
1043 for (const string& output_node : output_arg.output_nodes) { in InlineDirectFunctionCall()
1044 output_nodes.insert(output_node); in InlineDirectFunctionCall()
1062 if (IsIdentity(func_body_node) && output_nodes.count(node_name)) { in InlineDirectFunctionCall()
1348 absl::flat_hash_set<absl::string_view> output_nodes; in MaybeDeadOutputs() local
1350 for (const auto& output_node : output_expansion.output_nodes) { in MaybeDeadOutputs()
1351 output_nodes.insert(output_node); in MaybeDeadOutputs()
1366 if (output_nodes.find(node->name()) != output_nodes.end()) { in MaybeDeadOutputs()
[all …]
/external/tensorflow/tensorflow/cc/framework/
Dgradients.cc99 const std::unordered_set<int>& output_nodes);
193 const std::unordered_set<int>& output_nodes) { in GetStopBackpropNodes() argument
224 if (output_nodes.find(node_id) != output_nodes.end()) { in GetStopBackpropNodes()
241 for (int output_node : output_nodes) { in GetStopBackpropNodes()
266 std::unordered_set<int> output_nodes; in Initialize() local
267 output_nodes.reserve(outputs_.size()); in Initialize()
269 output_nodes.insert(outputs_[i].node()->id()); in Initialize()
273 GetStopBackpropNodes(reachable_nodes, output_nodes); in Initialize()
319 if (output_nodes.find(n->id()) != output_nodes.end()) { in Initialize()
/external/tensorflow/tensorflow/contrib/remote_fused_graph/pylib/python/ops/
Dremote_fused_graph_ops_test.py54 output_nodes = remote_fused_graph_ops.remote_fused_graph_execute(
59 self.assertEqual(2, len(output_nodes))
60 for output_node in output_nodes:
/external/libaom/libaom/av1/encoder/
Dml.c33 float *output_nodes = buf[buf_index]; in av1_nn_predict_c() local
42 output_nodes[node] = val; in av1_nn_predict_c()
45 input_nodes = output_nodes; in av1_nn_predict_c()
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.cc334 for (const string& output_node : output_arg.output_nodes) { in GrapplerFunctionItem()
619 absl::InlinedVector<string, 1> output_nodes; in MakeGrapplerFunctionItem() local
629 output_nodes.push_back(identity->name()); in MakeGrapplerFunctionItem()
635 /*output_nodes=*/std::move(output_nodes)}; in MakeGrapplerFunctionItem()
739 DCHECK(out_arg.output_nodes.size() == 1) in RemoveFunctionOutputs()
793 absl::flat_hash_set<absl::string_view> output_nodes; in MakeFunctionDef() local
795 for (const string& output_node : output_arg.output_nodes) { in MakeFunctionDef()
796 output_nodes.insert(output_node); in MakeFunctionDef()
807 if (output_nodes.find(node_name) != output_nodes.end()) { in MakeFunctionDef()
825 const string& output_node = output_arg.output_nodes[0]; in MakeFunctionDef()
[all …]
Dfunctions_test.cc261 EXPECT_EQ("y_output_node_0", item.output(0).output_nodes[0]); in TEST_F()
342 EXPECT_EQ("dx_output_node_0", item.output(0).output_nodes[0]); in TEST_F()
343 EXPECT_EQ("dy_output_node_0", item.output(1).output_nodes[0]); in TEST_F()
488 EXPECT_EQ("out_output_node_0", item.output(0).output_nodes[0]); in TEST_F()
540 EXPECT_EQ("out0_output_node_0", item.output(0).output_nodes[0]); in TEST_F()
541 EXPECT_EQ("arg2_output_node_0", item.output(1).output_nodes[0]); in TEST_F()
542 EXPECT_EQ("arg3_output_node_0", item.output(2).output_nodes[0]); in TEST_F()
603 EXPECT_EQ("o_output_node_0", item.output(0).output_nodes[0]); in TEST_F()
/external/tensorflow/tensorflow/tools/graph_transforms/
Dbackports.cc36 const std::set<string>& output_nodes, in BackportConcatV2Transform()
73 const std::set<string>& output_nodes, in BackportTensorArrayV3Transform() argument
119 const std::set<string>& output_nodes, in BackportTensorArrayV3Transform() argument
Dfuse_convolutions.cc47 const std::set<string>& output_nodes, in FuseResizePadAndConv()
96 const std::set<string>& output_nodes, in FuseResizeAndConv() argument
156 const std::set<string>& output_nodes, in FusePadAndConv() argument
Dfold_batch_norms.cc50 const std::set<string>& output_nodes, in FoldBatchNorms()
61 if (output_nodes.count(node.name())) { in FoldBatchNorms()
Dquantize_nodes.cc281 const std::set<string>& output_nodes, in RemoveRedundantQuantizations()
295 if (output_nodes.count(dequantize_node.name()) || in RemoveRedundantQuantizations()
404 const std::set<string>& output_nodes, in ConvertFakeQuantsToRequantize()
498 const std::set<string>& output_nodes, in MergeAdjacentRequantizes()
548 const std::set<string>& output_nodes, in HoistFakeQuants()
697 const std::set<string>& output_nodes, in QuantizeNodes()
Dfold_old_batch_norms.cc313 const std::set<string>& output_nodes, in FoldOldBatchNorms()
350 const std::set<string>& output_nodes, in FoldOldBatchNorms()
393 const std::set<string>& output_nodes, in FoldOldBatchNorms()
/external/autotest/server/site_tests/audio_AudioBasicHDMI/
Daudio_AudioBasicHDMI.py98 output_nodes, _ = audio_facade.get_selected_node_types()
100 if output_nodes == ['HEADPHONE'] or output_nodes == ['LINEOUT']:
/external/autotest/server/site_tests/audio_LeftRightInternalSpeaker/
Daudio_LeftRightInternalSpeaker.py110 output_nodes, _ = self.audio_facade.get_selected_node_types()
111 if output_nodes != ['INTERNAL_SPEAKER']:
114 'device' % output_nodes)
/external/autotest/client/common_lib/cros/cfm/
Dcras_node_collector_unittest.py23 output_nodes = (
44 mock_host.run_output.side_effect = [output_devices, output_nodes]
/external/autotest/server/site_tests/audio_AudioBasicBluetoothPlayback/
Daudio_AudioBasicBluetoothPlayback.py135 output_nodes, _ = self.audio_facade.get_plugged_node_types()
136 if 'HEADPHONE' in output_nodes or 'LINEOUT' in output_nodes:
/external/autotest/server/site_tests/audio_AudioBasicBluetoothRecord/
Daudio_AudioBasicBluetoothRecord.py139 output_nodes, _ = self.audio_facade.get_plugged_node_types()
141 if 'HEADPHONE' in output_nodes or 'LINEOUT' in output_nodes:
/external/tensorflow/tensorflow/lite/python/
Dconvert_test.py196 def _getGraphOpTypes(self, graphdef, output_nodes): argument
215 used_node_names = _bfs_for_reachable_nodes(output_nodes, name_to_input_name)
253 output_nodes=[op_hint._tensor_name_base(output.name)]),
280 output_nodes=[op_hint._tensor_name_base(output.name)]),
305 output_nodes=[op_hint._tensor_name_base(output.name)]),
381 output_nodes=[op_hint._tensor_name_base(output.name)]),
/external/autotest/client/cros/audio/
Dcras_utils.py233 output_nodes = []
241 output_nodes.append(node['Id'])
242 return (output_nodes, input_nodes)
/external/autotest/server/site_tests/audio_AudioBasicBluetoothPlaybackRecord/
Daudio_AudioBasicBluetoothPlaybackRecord.py155 output_nodes, _ = self.audio_facade.get_plugged_node_types()
157 if 'HEADPHONE' in output_nodes or 'LINEOUT' in output_nodes:
/external/tensorflow/tensorflow/python/autograph/impl/
Dconversion.py259 output_nodes = []
268 output_nodes.append(
296 output_nodes.append(class_def)
298 return output_nodes, class_name, class_namespace
/external/autotest/client/cros/chameleon/
Daudio_test_utils.py763 output_nodes, _ = audio_facade.get_plugged_node_types()
764 if 'LINEOUT' in output_nodes:
766 if 'HEADPHONE' in output_nodes:

12