| /frameworks/ml/nn/runtime/test/specs/ |
| D | rnn.mod.py | 19 input_size = 8 variable
23 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (batches, input_size))
24 weights = Input("weights", "TENSOR_FLOAT32", "{%d, %d}" % (units, input_size))
185 input_sequence_size = int(len(test_inputs) / input_size / batches)
190 input_begin = i * input_size
191 input_end = input_begin + input_size
|
| D | svdf.mod.py | 19 input_size = 3 variable
24 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (batches, input_size))
25 weights_feature = Input("weights_feature", "TENSOR_FLOAT32", "{%d, %d}" % (units, input_size))
126 input_sequence_size = int(len(test_inputs) / input_size / batches)
131 batch_start = i * input_size * batches
132 batch_end = batch_start + input_size * batches
|
| D | rnn_state.mod.py | 19 input_size = 8 variable
23 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (batches, input_size))
24 weights = Input("weights", "TENSOR_FLOAT32", "{%d, %d}" % (units, input_size))
|
| D | svdf_state.mod.py | 19 input_size = 3 variable
24 input = Input("input", "TENSOR_FLOAT32", "{%d, %d}" % (batches, input_size))
25 weights_feature = Input("weights_feature", "TENSOR_FLOAT32", "{%d, %d}" % (units, input_size))
|
| /frameworks/av/media/libaudioprocessing/tests/ |
| D | test-resampler.cpp | 208 size_t input_size; in main() local
218 input_size = info.frames * info.channels * sizeof(short); in main()
219 input_vaddr = malloc(input_size); in main()
230 input_size = channels * sizeof(int16_t) * input_frames; in main()
231 input_vaddr = malloc(input_size); in main()
243 size_t input_frames = input_size / input_framesize; in main()
250 input_size = input_frames * input_framesize; in main()
251 new_vaddr = malloc(input_size); in main()
|
| /frameworks/ml/nn/common/operations/ |
| D | RNNTest.cpp | 211 uint32_t input_size() const { return input_size_; } in input_size() function in android::nn::wrapper::BasicRNNOpModel
315 (rnn.input_size() * rnn.num_batches()); in TEST()
318 float* batch_start = rnn_input + i * rnn.input_size(); in TEST()
319 float* batch_end = batch_start + rnn.input_size(); in TEST()
321 rnn.SetInput(rnn.input_size(), batch_start, batch_end); in TEST()
|
| D | Activation.cpp | 76 uint32_t input_size = getNumberOfElements(inputShape) / batch_size; in softmaxFloat32() local
79 shapeIn4D.dimensions = {batch_size, 1, 1, input_size}; in softmaxFloat32()
167 uint32_t input_size = getNumberOfElements(inputShape) / batch_size; in softmaxQuant8() local
170 shapeIn4D.dimensions = {batch_size, 1, 1, input_size}; in softmaxQuant8()
|
| D | RNN.cpp | 85 const uint32_t input_size = input_->shape().dimensions[1]; in Eval() local
94 reinterpret_cast<float*>(input_->buffer) + b * input_size; in Eval()
114 for (uint32_t i = 0; i < input_size; i++) { in Eval()
|
| D | SVDFTest.cpp | 122 SVDFOpModel(uint32_t batches, uint32_t units, uint32_t input_size, in SVDFOpModel() argument
126 input_size_(input_size), in SVDFOpModel()
253 int input_size() const { return input_size_; } in input_size() function in android::nn::wrapper::SVDFOpModel
298 const int svdf_input_size = svdf.input_size(); in TEST()
|
| D | SVDF.cpp | 108 const int input_size = input_->shape().dimensions[1]; in Eval() local
121 const float* input_ptr_batch = reinterpret_cast<float *>(input_->buffer) + b * input_size; in Eval()
131 for (int j = 0; j < input_size; j++) { in Eval()
|
| /frameworks/base/media/mca/filterfw/jni/ |
| D | jni_native_program.cpp | 148 int input_size = 0; in Java_android_filterfw_core_NativeProgram_callNativeProcess() local
157 input_size = native_frame->Size(); in Java_android_filterfw_core_NativeProgram_callNativeProcess()
160 input_sizes[i] = input_size; in Java_android_filterfw_core_NativeProgram_callNativeProcess()
|
| /frameworks/ml/nn/common/ |
| D | OperationsUtils.cpp | 332 uint32_t input_size = getNumberOfElements(input); in fullyConnectedPrepare() local
334 uint32_t batch_size = input_size / getSizeOfDimension(weights, 1); in fullyConnectedPrepare()
337 NN_OPS_CHECK(getSizeOfDimension(weights, 1) * batch_size == input_size); in fullyConnectedPrepare()
|
| /frameworks/ml/nn/common/operations/internal/optimized/ |
| D | optimized_ops.h | 390 const int input_size = input_dims.strides[3]; in FullyConnectedAsGEMV() local
393 for (int k = 0; k < input_size; k += 64) { in FullyConnectedAsGEMV()
396 for (int k = 0; k < kPeel * input_size; k += 64) { in FullyConnectedAsGEMV()
410 for (; in <= input_size - 16; in += 16) { in FullyConnectedAsGEMV()
414 const uint8* filter_ptr = filter_data + in + (out + k) * input_size; in FullyConnectedAsGEMV()
445 for (; in <= input_size - 8; in += 8) { in FullyConnectedAsGEMV()
449 const uint8* filter_ptr = filter_data + in + (out + k) * input_size; in FullyConnectedAsGEMV()
469 if (in < input_size) { in FullyConnectedAsGEMV()
474 for (; in < input_size; in++) { in FullyConnectedAsGEMV()
475 int lane = (in + 8 - input_size) % 4; in FullyConnectedAsGEMV()
[all …]
|