Searched defs:weights_val (Results 1 – 7 of 7) sorted by relevance
242 int8_t weights_val = *shuffled_weights_ptr++; in ShuffledFullyConnected() local289 int8_t weights_val = shuffled_weights_ptr[16 * i + j]; in ShuffledFullyConnected() local
340 int8_t weights_val = input_to_gate_weights[row * n_input + col]; in PortableMatrixBatchVectorMultiplyAccumulateImpl() local392 int8_t weights_val = input_to_gate_weights[row * n_input + col]; in PortableMatrixBatchVectorMultiply() local421 int8_t weights_val = hidden_to_output_weights[row * n_hidden + col]; in PortableMatrixBatchVectorMultiply() local
329 int16_t weights_val = in LstmCell() local
298 …const auto weights_val = *(reinterpret_cast<T *>(weights_ptr + m * sizeof(T) + w * run_info.weight… in depthwise_loop_generic_fp() local462 …const auto weights_val = *(reinterpret_cast<TW *>(weights_ptr + w * run_info.weights_stride_y) + x… in depthwise_loop_multiplier1_quantized() local567 …const auto weights_val = *(reinterpret_cast<TW *>(weights_ptr + m * sizeof(T) + w * run_info.weigh… in depthwise_loop_generic_quantized() local
45 int8_t weights_val = input_to_gate_weights[row * n_input + col]; in MatrixBatchVectorMultiplyAccumulate() local
116 int16_t weights_val = in quantizedLstmStep() local
3965 def forward_backward(cuda, rnn, input_val, grad_output, weights_val, hx_val, grad_hy, argument