Searched refs:aux_input_ (Results 1 – 2 of 2) sorted by relevance
149 aux_input_ = GetInput(operation, operands, kAuxInputTensor); in BidirectionalSequenceLSTM()326 const bool has_aux_input = !IsNullInput(aux_input_); in Prepare()337 NN_CHECK_EQ(aux_input_->shape().dimensions[0], input_->shape().dimensions[0]); in Prepare()338 NN_CHECK_EQ(aux_input_->shape().dimensions[1], input_->shape().dimensions[1]); in Prepare()393 linkingMode == LinkingMode::PARALLEL_LINKING ? aux_input_ : input_; in Prepare()451 const bool has_aux_input = !IsNullInput(aux_input_); in Eval()461 const float* auxInput = GetOptionalBuffer<const float>(aux_input_); in Eval()463 bwInput = GetBuffer<const float>(aux_input_); in Eval()464 bwInputShape = aux_input_->shape(); in Eval()584 const _Float16* auxInput = GetOptionalBuffer<const _Float16>(aux_input_); in Eval()[all …]
165 const RunTimeOperandInfo* aux_input_; variable