Home
last modified time | relevance | path

Searched refs:batch_size (Results 1 – 15 of 15) sorted by relevance

/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/reference/
A Dsvdf.h44 for (int b = 0; b < batch_size; ++b) { in ApplyTimeWeightsBiasAndActivation()
54 batch_size * num_units, rank); in ApplyTimeWeightsBiasAndActivation()
57 tensor_utils::VectorBatchVectorAdd(bias_ptr, num_units, batch_size, in ApplyTimeWeightsBiasAndActivation()
160 const int batch_size = input_shape.Dims(0); in EvalFloatSVDF() local
173 std::fill_n(scratch_data, batch_size * num_filters, 0.0f); in EvalFloatSVDF()
182 for (int i = 0; i < batch_size * num_filters; ++i) { in EvalFloatSVDF()
201 const int batch_size = input_shape.Dims(0); in EvalHybridSVDF() local
213 std::fill_n(scratch, batch_size * num_filters, 0.0f); in EvalHybridSVDF()
220 for (int b = 0; b < batch_size; ++b) { in EvalHybridSVDF()
227 scaling_factors, batch_size, scratch, in EvalHybridSVDF()
[all …]
A Dgather.h52 int batch_size = 1; in Gather() local
54 batch_size *= input_shape.Dims(i); in Gather()
72 for (int batch = 0; batch < batch_size; ++batch) { in Gather()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/experimental/kernels/
A Dctc_beam_search_decoder.cc71 const int batch_size = SizeOfDimension(inputs, 1); in Prepare() local
77 TF_LITE_ENSURE_EQ(context, NumElements(sequence_length), batch_size); in Prepare()
102 log_probability_output_shape_array->data[0] = batch_size; in Prepare()
124 const int32_t batch_size = sequences.size(); in StoreAllDecodedSequences() local
159 for (int b = 0; b < batch_size; ++b) { in StoreAllDecodedSequences()
171 decoded_shape_data[0] = batch_size; in StoreAllDecodedSequences()
188 const int batch_size = SizeOfDimension(inputs, 1); in Eval() local
196 for (int i = 0; i < batch_size; ++i) { in Eval()
207 GetTensorData<float>(inputs) + t * batch_size * num_classes, batch_size, in Eval()
224 std::vector<std::vector<std::vector<int>>> best_paths(batch_size); in Eval()
[all …]
A Dctc_decoder.h43 CTCDecoder(int num_classes, int batch_size, bool merge_repeated) in CTCDecoder() argument
46 batch_size_(batch_size), in CTCDecoder()
60 int batch_size() { return batch_size_; } in batch_size() function
74 CTCGreedyDecoder(int num_classes, int batch_size, bool merge_repeated) in CTCGreedyDecoder() argument
75 : CTCDecoder(num_classes, batch_size, merge_repeated) {} in CTCGreedyDecoder()
A Dctc_beam_search.h88 BaseBeamScorer<CTCBeamState>* scorer, int batch_size = 1,
90 : CTCDecoder(num_classes, batch_size, merge_repeated), in CTCDecoder() argument
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/
A Dkernel_utils.h41 int input_size, int num_units, int batch_size,
52 int batch_size, int output_batch_leading_dim,
70 int num_units, int batch_size, int output_batch_leading_dim,
83 int batch_size, int output_batch_leading_dim,
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/cmsis_nn/
A Dsvdf.cc78 for (int b = 0; b < batch_size; ++b) { in ApplyTimeWeightsBiasAndActivation()
95 for (int i = 0; i < batch_size; ++i) { in ApplyTimeWeightsBiasAndActivation()
104 for (int i = 0; i < batch_size * num_units; ++i) { in ApplyTimeWeightsBiasAndActivation()
110 for (int b = 0; b < batch_size; ++b) { in ApplyTimeWeightsBiasAndActivation()
123 for (int b = 0; b < batch_size; ++b) { in ApplyTimeWeightsBiasAndActivation()
140 const int batch_size = input->dims->data[0]; in EvalFloatSVDF() local
168 state_ptr + batch_size * num_filters * memory_size; in EvalFloatSVDF()
187 for (int i = 0; i < batch_size; ++i) { in EvalFloatSVDF()
314 const int batch_size = input->dims->data[0]; in Prepare() local
395 context, batch_size * num_filters * sizeof(int32_t), in Prepare()
[all …]
A Ddepthwise_conv.cc114 const int batch_size = MatchingDim(input_shape, 0, output_shape, 0); in Prepare() local
116 TFLITE_DCHECK_EQ(batch_size, 1); /* Only batch = 1 is supported */ in Prepare()
119 input_dims.n = batch_size; in Prepare()
131 output_dims.n = batch_size; in Prepare()
189 const int batch_size = MatchingDim(input_shape, 0, output_shape, 0); in EvalQuantizedPerChannel() local
197 input_dims.n = batch_size; in EvalQuantizedPerChannel()
215 output_dims.n = batch_size; in EvalQuantizedPerChannel()
A Dconv.cc170 const int batch_size = MatchingDim(input_shape, 0, output_shape, 0); in EvalQuantizedPerChannel() local
180 input_dims.n = batch_size; in EvalQuantizedPerChannel()
201 output_dims.n = batch_size; in EvalQuantizedPerChannel()
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/
A Dsvdf_common.cc202 for (int b = 0; b < batch_size; ++b) { in ApplyTimeWeightsBiasAndActivation()
219 for (int i = 0; i < batch_size; ++i) { in ApplyTimeWeightsBiasAndActivation()
228 for (int i = 0; i < batch_size * num_units; ++i) { in ApplyTimeWeightsBiasAndActivation()
234 for (int b = 0; b < batch_size; ++b) { in ApplyTimeWeightsBiasAndActivation()
247 for (int b = 0; b < batch_size; ++b) { in ApplyTimeWeightsBiasAndActivation()
264 const int batch_size = input->dims->data[0]; in EvalFloatSvdfReference() local
292 state_ptr + batch_size * num_filters * memory_size; in EvalFloatSvdfReference()
311 for (int i = 0; i < batch_size; ++i) { in EvalFloatSvdfReference()
359 const int batch_size = input->dims->data[0]; in PrepareSvdf() local
441 context, batch_size * num_filters * sizeof(int32_t), in PrepareSvdf()
[all …]
/AliOS-Things-master/solutions/tflite_micro_speech_demo/micro_speech/train/speech_commands/
A Dtrain.py236 FLAGS.batch_size, 0, model_settings, FLAGS.background_frequency,
267 for i in xrange(0, set_size, FLAGS.batch_size):
269 audio_processor.get_data(FLAGS.batch_size, i, model_settings, 0.0,
281 batch_size = min(FLAGS.batch_size, set_size - i)
282 total_accuracy += (validation_accuracy * batch_size) / set_size
304 for i in xrange(0, set_size, FLAGS.batch_size):
306 FLAGS.batch_size, i, model_settings, 0.0, 0.0, 0, 'testing', sess)
314 batch_size = min(FLAGS.batch_size, set_size - i)
315 total_accuracy += (test_accuracy * batch_size) / set_size
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/integer_ops/
A Dtranspose_conv.h38 const int batch_size = MatchingDim(input_shape, 0, output_shape, 0); in TransposeConvV2() local
72 std::fill_n(scratch_data, output_offset * batch_size, static_cast<int32>(0)); in TransposeConvV2()
73 for (int i = 0; i < batch_size; ++i) { in TransposeConvV2()
98 optimized_ops::BiasAdd(scratch_data_p, bias_data, batch_size, output_height, in TransposeConvV2()
/AliOS-Things-master/components/py_engine/engine/py/
A Dmakeqstrdefs.py60 batch_size = (len(sources) + cpus - 1) // cpus
61 chunks = [sources[i : i + batch_size] for i in range(0, len(sources), batch_size or 1)]
/AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/
A Doptimized_ops.h1351 const int batch_size = input_shape.Dims(0); in HybridConv() local
1451 const int batch_size = input_shape.Dims(0); in HybridConvPerChannel() local
1457 batch_size); in HybridConvPerChannel()
1658 const int batch_size = output_shape.Dims(0); in DepthToSpace() local
1663 for (int batch = 0; batch < batch_size; ++batch) { in DepthToSpace()
1699 const int batch_size = input_shape.Dims(0); in SpaceToDepth() local
3709 const int batch_size = variable
3720 batch_size);
5252 for (int n = 0; n < batch_size; ++n) { in BiasAdd()
5311 for (int i = 0; i < batch_size; ++i) { in TransposeConvV2()
[all …]
/AliOS-Things-master/solutions/tflite_micro_speech_demo/micro_speech/train/
A Dtrain_micro_speech_model.ipynb265 "--batch_size={BATCH_SIZE} \\\n",

Completed in 55 milliseconds