| /frameworks/ml/nn/runtime/test/specs/ |
| H A 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
|
| H A 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
|
| H A 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))
|
| H A 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/ml/nn/common/operations/ |
| H A D | RNNTest.cpp | 211 uint32_t input_size() const { return input_size_; } function in class:android::nn::wrapper::BasicRNNOpModel
315 (rnn.input_size() * rnn.num_batches());
318 float* batch_start = rnn_input + i * rnn.input_size();
319 float* batch_end = batch_start + rnn.input_size();
321 rnn.SetInput(rnn.input_size(), batch_start, batch_end);
|
| H A D | RNN.cpp | 85 const uint32_t input_size = input_->shape().dimensions[1]; local
94 reinterpret_cast<float*>(input_->buffer) + b * input_size;
114 for (uint32_t i = 0; i < input_size; i++) {
|
| H A D | SVDF.cpp | 108 const int input_size = input_->shape().dimensions[1]; local
121 const float* input_ptr_batch = reinterpret_cast<float *>(input_->buffer) + b * input_size;
131 for (int j = 0; j < input_size; j++) {
|
| H A D | Activation.cpp | 76 uint32_t input_size = getNumberOfElements(inputShape) / batch_size; local
79 shapeIn4D.dimensions = {batch_size, 1, 1, input_size};
167 uint32_t input_size = getNumberOfElements(inputShape) / batch_size; local
170 shapeIn4D.dimensions = {batch_size, 1, 1, input_size};
|
| H A D | SVDFTest.cpp | 122 SVDFOpModel(uint32_t batches, uint32_t units, uint32_t input_size, argument
126 input_size_(input_size),
253 int input_size() const { return input_size_; } function in class:android::nn::wrapper::SVDFOpModel
275 SVDFOpModel svdf(/*batches=*/2, /*units=*/4, /*input_size=*/3,
298 const int svdf_input_size = svdf.input_size();
|
| /frameworks/av/media/libaudioprocessing/tests/ |
| H A D | test-resampler.cpp | 208 size_t input_size; local
218 input_size = info.frames * info.channels * sizeof(short);
219 input_vaddr = malloc(input_size);
230 input_size = channels * sizeof(int16_t) * input_frames;
231 input_vaddr = malloc(input_size);
243 size_t input_frames = input_size / input_framesize;
250 input_size = input_frames * input_framesize;
251 new_vaddr = malloc(input_size);
|
| /frameworks/base/media/mca/filterfw/jni/ |
| H A D | jni_native_program.cpp | 148 int input_size = 0; local
157 input_size = native_frame->Size();
160 input_sizes[i] = input_size;
|
| /frameworks/ml/nn/common/ |
| H A D | OperationsUtils.cpp | 332 uint32_t input_size = getNumberOfElements(input);
334 uint32_t batch_size = input_size / getSizeOfDimension(weights, 1);
337 NN_OPS_CHECK(getSizeOfDimension(weights, 1) * batch_size == input_size);
|
| /frameworks/ml/nn/common/operations/internal/optimized/ |
| H A D | optimized_ops.h | 390 const int input_size = input_dims.strides[3]; local
393 for (int k = 0; k < input_size; k += 64) {
396 for (int k = 0; k < kPeel * input_size; k += 64) {
410 for (; in <= input_size - 16; in += 16) {
414 const uint8* filter_ptr = filter_data + in + (out + k) * input_size;
445 for (; in <= input_size - 8; in += 8) {
449 const uint8* filter_ptr = filter_data + in + (out + k) * input_size;
469 if (in < input_size) {
474 for (; in < input_size; in++) {
475 int lane = (in + 8 - input_size)
[all...] |