Lines Matching refs:inputShape
68 const Shape& inputShape = input->shape(); in Prepare() local
71 hiddenStateShape->type = inputShape.type; in Prepare()
75 outputShape->type = inputShape.type; in Prepare()
116 bool RNN::RNNStep(const T* inputData, const Shape& inputShape, const T* hiddenStateInputData, in RNNStep() argument
124 return RNNStep<T>(inputData, inputShape, /*auxInputData=*/nullptr, /*auxInputShape=*/dummyShape, in RNNStep()
136 bool RNN::RNNStep(const T* inputData, const Shape& inputShape, const T* auxInputData, in RNNStep() argument
145 const uint32_t batch_size = inputShape.dimensions[0]; in RNNStep()
147 const uint32_t input_size = inputShape.dimensions[1]; in RNNStep()
223 template bool RNN::RNNStep<_Float16>(const _Float16* inputData, const Shape& inputShape,
229 template bool RNN::RNNStep<_Float16>(const _Float16* inputData, const Shape& inputShape,
239 template bool RNN::RNNStep<float>(const float* inputData, const Shape& inputShape,
245 template bool RNN::RNNStep<float>(const float* inputData, const Shape& inputShape,