Searched refs:aux_input_size (Results 1 – 2 of 2) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | RNN.cpp | 151 uint32_t aux_input_size = 0; in RNNStep() local
155 aux_input_size = auxInputShape.dimensions[1]; in RNNStep()
166 aux_input_ptr_batch = auxInputData + b * aux_input_size; in RNNStep()
|
| /aosp12/hardware/interfaces/neuralnetworks/1.3/ |
| H A D | types.hal | 2552 * A 3-D tensor of shape [max_time, batch_size, aux_input_size],
2554 * “aux_input_size” is the size of the auxiliary input. Optional. See
2558 * A 2-D tensor of shape [fw_num_units, aux_input_size].
2561 * A 2-D tensor of shape [fw_num_units, aux_input_size].
2564 * A 2-D tensor of shape [fw_num_units, aux_input_size].
2567 * A 2-D tensor of shape [fw_num_units, aux_input_size].
2570 * A 2-D tensor of shape [bw_num_units, aux_input_size].
2573 * A 2-D tensor of shape [bw_num_units, aux_input_size].
2576 * A 2-D tensor of shape [bw_num_units, aux_input_size].
2579 * A 2-D tensor of shape [bw_num_units, aux_input_size].
|