• Home
  • Raw
  • Download

Lines Matching refs:Shape

65 void transposeFirstTwoDims(const T* input, const Shape& inputShape, T* output) {  in transposeFirstTwoDims()
80 Shape removeFirstDim(const Shape& input) { in removeFirstDim()
81 Shape output = input; in removeFirstDim()
123 Shape inputShape = context->getInputShape(kInputTensor); in executeTyped()
126 Shape fwWeightsShape = context->getInputShape(kFwWeightsTensor); in executeTyped()
128 Shape fwRecurrentWeightsShape = context->getInputShape(kFwRecurrentWeightsTensor); in executeTyped()
133 Shape bwWeightsShape = context->getInputShape(kBwWeightsTensor); in executeTyped()
135 Shape bwRecurrentWeightsShape = context->getInputShape(kBwRecurrentWeightsTensor); in executeTyped()
154 Shape auxInputShape = context->getInputShape(kAuxInputTensor); in executeTyped()
155 Shape fwAuxWeightsShape = context->getInputShape(kFwAuxWeightsTensor); in executeTyped()
156 Shape bwAuxWeightsShape = context->getInputShape(kBwAuxWeightsTensor); in executeTyped()
163 Shape fwOutputShape = context->getOutputShape(kFwOutputTensor); in executeTyped()
165 Shape bwOutputShape; in executeTyped()
227 Shape fixedTimeInputShape = removeFirstDim(inputShape); in executeTyped()
228 Shape fixedTimeAuxInputShape = auxInputShape; in executeTyped()
367 Shape input = context->getInputShape(kInputTensor); in prepare()
368 Shape fwWeights = context->getInputShape(kFwWeightsTensor); in prepare()
369 Shape fwRecurrentWeights = context->getInputShape(kFwRecurrentWeightsTensor); in prepare()
370 Shape fwBias = context->getInputShape(kFwBiasTensor); in prepare()
371 Shape fwHiddenState = context->getInputShape(kFwHiddenStateTensor); in prepare()
372 Shape bwWeights = context->getInputShape(kBwWeightsTensor); in prepare()
373 Shape bwRecurrentWeights = context->getInputShape(kBwRecurrentWeightsTensor); in prepare()
374 Shape bwBias = context->getInputShape(kBwBiasTensor); in prepare()
375 Shape bwHiddenState = context->getInputShape(kBwHiddenStateTensor); in prepare()
377 Shape auxInput = context->getInputShape(kAuxInputTensor); in prepare()
378 Shape fwAuxWeights = context->getInputShape(kFwAuxWeightsTensor); in prepare()
379 Shape bwAuxWeights = context->getInputShape(kBwAuxWeightsTensor); in prepare()
438 Shape fwOutput = context->getOutputShape(kFwOutputTensor); in prepare()
445 Shape bwOutput = context->getOutputShape(kBwOutputTensor); in prepare()