• Home
  • Raw
  • Download

Lines Matching refs:input_

82     input_ = GetInput(operation, operands, kInputTensor);  in BidirectionalSequenceLSTM()
176 if (input_->type == OperandType::TENSOR_FLOAT32) { in BidirectionalSequenceLSTM()
253 if (input_->type == OperandType::TENSOR_FLOAT32) { in Prepare()
267 NN_CHECK(NumDimensions(input_) == 3); in Prepare()
268 const uint32_t max_time = SizeOfDimension(input_, params_.time_major ? 0 : 1); in Prepare()
269 const uint32_t n_batch = SizeOfDimension(input_, params_.time_major ? 1 : 0); in Prepare()
270 const uint32_t n_fw_input = SizeOfDimension(input_, 2); in Prepare()
288 input_, fw_input_to_input_weights_, fw_input_to_forget_weights_, 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()
341 int n_aux_input = SizeOfDimension(input_, 2); in Prepare()
382 const Shape& inputShape = input_->shape(); in Prepare()
392 linkingMode == LinkingMode::PARALLEL_LINKING ? aux_input_ : input_; in Prepare()
442 std::vector<uint32_t> fw_output_dims = input_->shape().dimensions; in Eval()
456 switch (input_->type) { in Eval()
458 const float* bwInput = GetBuffer<const float>(input_); in Eval()
459 Shape bwInputShape = input_->shape(); in Eval()
485 params_, GetBuffer<const float>(input_), input_->shape(), in Eval()
581 const _Float16* bwInput = GetBuffer<const _Float16>(input_); in Eval()
582 Shape bwInputShape = input_->shape(); in Eval()
609 params_, GetBuffer<const _Float16>(input_), input_->shape(), in Eval()
706 LOG(ERROR) << "Unsupported data type: " << static_cast<int>(input_->type); in Eval()