Searched refs:input_ptr_batch (Results 1 – 2 of 2) sorted by relevance
115 const float* input_ptr_batch = inputData + b * class_size; in EvalFloat32() local118 if (Eigen::numext::isfinite(input_ptr_batch[j])) { in EvalFloat32()119 max = std::max(max, input_ptr_batch[j]); in EvalFloat32()127 if (Eigen::numext::isfinite(static_cast<float>(input_ptr_batch[j]))) { in EvalFloat32()128 total += exp(static_cast<double>(input_ptr_batch[j]) - batch_max); in EvalFloat32()
166 const T* input_ptr_batch = inputData + b * input_size; in RNNStep() local190 output_ptr_batch[o] += input_ptr_batch[i] * input_weights_ptr[i]; in RNNStep()