/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...] |