Searched refs:aux_input_ (Results 1 – 2 of 2) sorted by relevance
148 aux_input_ = GetInput(operation, operands, kAuxInputTensor); in BidirectionalSequenceLSTM()325 const bool has_aux_input = !IsNullInput(aux_input_); in Prepare()336 NN_CHECK_EQ(aux_input_->shape().dimensions[0], input_->shape().dimensions[0]); in Prepare()337 NN_CHECK_EQ(aux_input_->shape().dimensions[1], input_->shape().dimensions[1]); in Prepare()392 linkingMode == LinkingMode::PARALLEL_LINKING ? aux_input_ : input_; in Prepare()450 const bool has_aux_input = !IsNullInput(aux_input_); in Eval()460 const float* auxInput = GetOptionalBuffer<const float>(aux_input_); in Eval()462 bwInput = GetBuffer<const float>(aux_input_); in Eval()463 bwInputShape = aux_input_->shape(); in Eval()583 const _Float16* auxInput = GetOptionalBuffer<const _Float16>(aux_input_); in Eval()[all …]
162 const RunTimeOperandInfo* aux_input_; variable