• Home
  • Raw
  • Download

Lines Matching refs:params_

139     params_.activation = static_cast<TfLiteFusedActivation>(  in BidirectionalSequenceLSTM()
142 params_.cell_clip = getScalarData<float>(*GetInput(operation, operands, kCellClipParam)); in BidirectionalSequenceLSTM()
143 params_.proj_clip = getScalarData<float>(*GetInput(operation, operands, kProjClipParam)); in BidirectionalSequenceLSTM()
145 params_.cell_clip = static_cast<float>( in BidirectionalSequenceLSTM()
147 params_.proj_clip = static_cast<float>( in BidirectionalSequenceLSTM()
150 params_.merge_outputs = getScalarData<bool>(*GetInput(operation, operands, kMergeOutputsParam)); in BidirectionalSequenceLSTM()
151 params_.time_major = getScalarData<bool>(*GetInput(operation, operands, kTimeMajorParam)); in BidirectionalSequenceLSTM()
152 params_.use_layer_norm = !IsNullInput(fw_input_layer_norm_weights_); in BidirectionalSequenceLSTM()
155 if (!params_.merge_outputs) { in BidirectionalSequenceLSTM()
166 const uint32_t max_time = SizeOfDimension(input_, params_.time_major ? 0 : 1); in Prepare()
167 const uint32_t n_batch = SizeOfDimension(input_, params_.time_major ? 1 : 0); in Prepare()
188 fw_output_layer_norm_weights_, n_input, n_fw_output, n_fw_cell, &params_)) { in Prepare()
225 fwOutputShape->dimensions[0] = params_.time_major ? max_time : n_batch; in Prepare()
226 fwOutputShape->dimensions[1] = params_.time_major ? n_batch : max_time; in Prepare()
227 fwOutputShape->dimensions[2] = params_.merge_outputs ? n_fw_output + n_bw_output : n_fw_output; in Prepare()
239 bw_output_layer_norm_weights_, n_input, n_bw_output, n_bw_cell, &params_)) { in Prepare()
243 if (!params_.merge_outputs) { in Prepare()
248 bwOutputShape->dimensions[0] = params_.time_major ? max_time : n_batch; in Prepare()
249 bwOutputShape->dimensions[1] = params_.time_major ? n_batch : max_time; in Prepare()
253 if (params_.use_cifg) { in Prepare()
283 params_, GetBuffer<const float>(input_), input_->shape(), in Eval()
315 GetBuffer<float>(fw_output_), fw_scratch_buffer.data(), params_.time_major, in Eval()
321 params_, GetBuffer<const float>(input_), input_->shape(), in Eval()
353 params_.merge_outputs ? GetBuffer<float>(fw_output_) + n_fw_output_elements in Eval()
355 bw_scratch_buffer.data(), params_.time_major, kBackwardSequence); in Eval()
356 if (params_.merge_outputs) { in Eval()
369 params_, GetBuffer<const _Float16>(input_), input_->shape(), in Eval()
401 GetBuffer<_Float16>(fw_output_), fw_scratch_buffer.data(), params_.time_major, in Eval()
407 params_, GetBuffer<const _Float16>(input_), input_->shape(), in Eval()
439 params_.merge_outputs ? GetBuffer<_Float16>(fw_output_) + n_fw_output_elements in Eval()
441 bw_scratch_buffer.data(), params_.time_major, kBackwardSequence); in Eval()
442 if (params_.merge_outputs) { in Eval()