| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/experimental/kernels/ |
| A D | ctc_decoder.h | 43 CTCDecoder(int num_classes, int batch_size, bool merge_repeated) 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()
|
| A D | ctc_beam_search_decoder.cc | 71 const int batch_size = SizeOfDimension(inputs, 1); in Prepare() local
124 const int32_t batch_size = sequences.size(); in StoreAllDecodedSequences() local
188 const int batch_size = SizeOfDimension(inputs, 1); in Eval() local
|
| A D | ctc_beam_search.h | 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/reference/ |
| A D | svdf.h | 38 int batch_size, int memory_size, int num_filters, int num_units, int rank, in ApplyTimeWeightsBiasAndActivation()
160 const int batch_size = input_shape.Dims(0); in EvalFloatSVDF() local
201 const int batch_size = input_shape.Dims(0); in EvalHybridSVDF() local
|
| A D | gather.h | 52 int batch_size = 1; in Gather() local
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/cmsis_nn/ |
| A D | depthwise_conv.cc | 114 const int batch_size = MatchingDim(input_shape, 0, output_shape, 0); in Prepare() local
189 const int batch_size = MatchingDim(input_shape, 0, output_shape, 0); in EvalQuantizedPerChannel() local
|
| A D | svdf.cc | 72 int batch_size, int memory_size, int num_filters, int num_units, int rank, in ApplyTimeWeightsBiasAndActivation()
140 const int batch_size = input->dims->data[0]; in EvalFloatSVDF() local
314 const int batch_size = input->dims->data[0]; in Prepare() local
|
| A D | conv.cc | 170 const int batch_size = MatchingDim(input_shape, 0, output_shape, 0); in EvalQuantizedPerChannel() local
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/micro/kernels/ |
| A D | svdf_common.cc | 196 int batch_size, int memory_size, int num_filters, int num_units, int rank, in ApplyTimeWeightsBiasAndActivation()
264 const int batch_size = input->dims->data[0]; in EvalFloatSvdfReference() local
359 const int batch_size = input->dims->data[0]; in PrepareSvdf() local
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/integer_ops/ |
| A D | transpose_conv.h | 38 const int batch_size = MatchingDim(input_shape, 0, output_shape, 0); in TransposeConvV2() local
|
| /AliOS-Things-master/components/ai_agent/src/engine/tflite-micro/tensorflow/lite/kernels/internal/optimized/ |
| A D | optimized_ops.h | 1351 const int batch_size = input_shape.Dims(0); in HybridConv() local
1451 const int batch_size = input_shape.Dims(0); in HybridConvPerChannel() local
1658 const int batch_size = output_shape.Dims(0); in DepthToSpace() local
1699 const int batch_size = input_shape.Dims(0); in SpaceToDepth() local
3709 const int batch_size = variable
5249 void BiasAdd(T* im_data, const T* bias_data, const int batch_size, in BiasAdd()
5279 const int batch_size = MatchingDim(input_shape, 0, output_shape, 0); in TransposeConvV2() local
5505 const int batch_size = MatchingDim(input_shape, 0, output_shape, 0); in TransposeConvV2() local
|