Home
last modified time | relevance | path

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

12

/external/libaom/libaom/av1/encoder/
Dml.c30 void av1_nn_predict_c(const float *input_nodes, in av1_nn_predict_c() argument
49 val += layer_weights[node * num_input_nodes + i] * input_nodes[i]; in av1_nn_predict_c()
55 input_nodes = output_nodes; in av1_nn_predict_c()
65 val += layer_weights[node * num_input_nodes + i] * input_nodes[i]; in av1_nn_predict_c()
122 const float *input_nodes = feature; in av1_nn_predict_v2() local
128 input_nodes = nn_fc_forward(input_nodes, nn_config->layer + i); in av1_nn_predict_v2()
134 input_nodes = nn_fc_forward(input_nodes, nn_config->layer + num_layers); in av1_nn_predict_v2()
137 memcpy(output, input_nodes, sizeof(*input_nodes) * nn_config->num_logits); in av1_nn_predict_v2()
/external/tensorflow/tensorflow/c/
Dc_api_function.cc53 std::unordered_map<const Node*, std::vector<int>>* input_nodes) in ProcessInputs() argument
70 const auto& iter = input_nodes->find(node); in ProcessInputs()
71 if (iter == input_nodes->end()) { in ProcessInputs()
72 input_nodes->insert({node, {idx}}); in ProcessInputs()
112 const std::unordered_map<const Node*, std::vector<int>>& input_nodes, in ComputeBodyNodes() argument
117 const auto& iter = input_nodes.find(node); in ComputeBodyNodes()
118 if (iter == input_nodes.end()) { in ComputeBodyNodes()
162 std::unordered_map<const Node*, std::vector<int>> input_nodes; in TF_GraphToFunctionWithControlOutputs() local
164 &input_tensors, &input_nodes); in TF_GraphToFunctionWithControlOutputs()
194 fn_body, fn_name, num_opers, opers, input_nodes, &body_nodes); in TF_GraphToFunctionWithControlOutputs()
/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/tensorflow/tensorflow/tools/graph_transforms/
Dstrip_unused_nodes.cc120 std::set<string> input_nodes; in StripUnusedNodes() local
123 input_nodes.insert(NodeNameFromInput(input)); in StripUnusedNodes()
141 if (input_nodes.count(current_input)) { in StripUnusedNodes()
169 if (input_nodes.count(node.name())) { in StripUnusedNodes()
Dbackports.cc34 [](const NodeMatch& match, const std::set<string>& input_nodes, in BackportConcatV2Transform()
71 const std::set<string>& input_nodes, in BackportTensorArrayV3Transform() argument
117 [](const NodeMatch& match, const std::set<string>& input_nodes, in BackportTensorArrayV3Transform() argument
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()
Dfuse_convolutions.cc45 [](const NodeMatch& match, const std::set<string>& input_nodes, in FuseResizePadAndConv()
94 [](const NodeMatch& match, const std::set<string>& input_nodes, in FuseResizeAndConv() argument
154 [](const NodeMatch& match, const std::set<string>& input_nodes, in FusePadAndConv() argument
Dfold_old_batch_norms.cc311 const std::set<string>& input_nodes, in FoldOldBatchNorms()
348 const std::set<string>& input_nodes, in FoldOldBatchNorms()
391 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()
Dfold_batch_norms.cc48 [](const NodeMatch& match, const std::set<string>& input_nodes, in FoldBatchNorms()
Dremove_nodes.cc70 const NodeMatch& match, const std::set<string>& input_nodes, in RemoveNodes()
Dflatten_atrous.cc50 [](const NodeMatch& match, const std::set<string>& input_nodes, in FlattenAtrousConv()
/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/
Dauto_parallel.cc163 std::vector<const NodeDef*> input_nodes; in Initialize() local
167 {}, &input_nodes)); in Initialize()
169 LOG(INFO) << "Number of input nodes: " << input_nodes.size(); in Initialize()
181 for (const auto& input_node : input_nodes) { in Initialize()
Ddependency_optimizer.cc188 const NodeDef& node, const std::vector<NodeDef*>& input_nodes, in BypassingNodeIsBeneficial() argument
203 input_nodes[0]->device() != node.device()) || in BypassingNodeIsBeneficial()
214 for (NodeDef* input_node : input_nodes) { in BypassingNodeIsBeneficial()
225 for (NodeDef* input_node : input_nodes) { in BypassingNodeIsBeneficial()
377 std::vector<NodeDef*> input_nodes; in OptimizeNode() local
384 input_nodes.push_back(input_node); in OptimizeNode()
390 if (!BypassingNodeIsBeneficial(*node, input_nodes, output_nodes)) { in OptimizeNode()
401 const NodeDef* input = input_nodes[i]; in OptimizeNode()
Ddependency_optimizer.h50 const NodeDef& node, const std::vector<NodeDef*>& input_nodes,
/external/tensorflow/tensorflow/core/distributed_runtime/
Dcluster_function_library_runtime.cc57 std::vector<Node*> input_nodes; in ConstructFunctionGraph() local
58 input_nodes.reserve(sig.input_arg_size()); in ConstructFunctionGraph()
87 input_nodes.push_back(input_node); in ConstructFunctionGraph()
108 for (size_t i = 0; i < input_nodes.size(); ++i) { in ConstructFunctionGraph()
109 g.AddEdge(input_nodes[i], 0, function_node, i); in ConstructFunctionGraph()
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
Dquantization_config.cc47 std::vector<std::string> input_nodes = absl::StrSplit(node_names, ','); in ParseInputNodeQuantSpecs() local
78 return GetInputNodeQuantSpecs(input_nodes, node_mins, node_maxs, final_type, in ParseInputNodeQuantSpecs()
/external/tensorflow/tensorflow/python/framework/
Dgraph_util_impl.py316 input_nodes = input_graph.node
318 for node in input_nodes:
323 for node in input_nodes:
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dchoose_fastest_dataset_op.cc180 std::vector<Node*> input_nodes; in AsGraphDefInternal() local
181 input_nodes.reserve(inputs_.size()); in AsGraphDefInternal()
185 input_nodes.push_back(input_node); in AsGraphDefInternal()
190 this, {}, {std::make_pair(0, input_nodes)}, in AsGraphDefInternal()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc339 std::vector<Node*> input_nodes; in CreateTRTNode() local
409 input_nodes.push_back(CHECK_NOTNULL(input_node)); in CreateTRTNode()
522 VLOG(1) << "input_nodes size = " << input_nodes.size(); in CreateTRTNode()
523 for (int i = 0; i < input_nodes.size(); ++i) { in CreateTRTNode()
524 Node* n = CHECK_NOTNULL(input_nodes[i]); in CreateTRTNode()
/external/autotest/client/cros/audio/
Dcras_utils.py242 input_nodes = []
247 input_nodes.append(node['Id'])
250 return (output_nodes, input_nodes)
/external/adhd/cras/client/cras_tests/src/
Dcontrol.rs99 ListInputNodes => print_nodes(cras_client.input_nodes()), in control()
/external/adhd/cras/client/libcras/src/
Dcras_shm.rs554 input_nodes: VolatileSlice<'a>, field
606 input_nodes: vslice_from_addr!(addr, input_nodes), in try_new()
718 pub fn input_nodes(&self) -> impl Iterator<Item = CrasIonodeInfo> { in input_nodes() method
721 self.input_nodes.copy_to(&mut nodes); in input_nodes()

12