Lines Matching refs:input_
55 input_ = GetInput(operation, operands, kInputTensor); in LSTMCell()
92 if (input_->type == OperandType::TENSOR_FLOAT32) { in LSTMCell()
133 const RunTimeOperandInfo* input_, const RunTimeOperandInfo* input_to_input_weights, in CheckInputTensorDimensions() argument
337 if (input_->type == OperandType::TENSOR_FLOAT32) { in Prepare()
347 NN_CHECK(NumDimensions(input_) > 1); in Prepare()
348 const uint32_t n_batch = SizeOfDimension(input_, 0); in Prepare()
349 const uint32_t n_input = SizeOfDimension(input_, 1); in Prepare()
361 input_, input_to_input_weights_, input_to_forget_weights_, input_to_cell_weights_, in Prepare()
373 const Shape& inputShape = input_->shape(); in Prepare()
991 switch (input_->type) { in Eval()
993 LSTMEvalFloat32(params_, GetBuffer<const float>(input_), input_->shape(), in Eval()
1028 LSTMEvalFloat16(params_, GetBuffer<const _Float16>(input_), input_->shape(), in Eval()
1064 LOG(ERROR) << "Unsupported data type: " << static_cast<int>(input_->type); in Eval()