• Home
  • Raw
  • Download

Lines Matching refs:Shape

75             Shape filterShape = context->getInputShape(kFilterTensor);  in initialize()
126 bool transposeConvNhwc(const float* inputData, const Shape& inputShape, const float* filterData, in transposeConvNhwc()
127 const Shape& filterShape, const float* biasData, const Shape& biasShape, in transposeConvNhwc()
129 const Shape& outputShape) { in transposeConvNhwc()
183 bool transposeConvNhwc(const T* inputData, const Shape& inputShape, const T* filterData, in transposeConvNhwc()
184 const Shape& filterShape, const int32_t* biasData, const Shape& biasShape, in transposeConvNhwc()
185 const TransposeConv2dParam& param, T* outputData, const Shape& outputShape) { in transposeConvNhwc()
277 bool transposeConvNhwc(const _Float16* inputData, const Shape& inputShape, in transposeConvNhwc()
278 const _Float16* filterData, const Shape& filterShape, in transposeConvNhwc()
279 const _Float16* biasData, const Shape& biasShape, in transposeConvNhwc()
281 const Shape& outputShape) { in transposeConvNhwc()
301 bool transposeConv(const T_Input* inputData, const Shape& inputShape, const T_Filter* filterData, in transposeConv()
302 const Shape& filterShape, const T_Bias* biasData, const Shape& biasShape, in transposeConv()
304 const Shape& outputShape) { in transposeConv()
317 bool transposeConvQuant8PerChannelNhwc(const T* inputData, const Shape& inputShape, in transposeConvQuant8PerChannelNhwc()
318 const int8_t* filterData, const Shape& filterShape, in transposeConvQuant8PerChannelNhwc()
320 const Shape& biasShape, const TransposeConv2dParam& param, in transposeConvQuant8PerChannelNhwc()
321 T* outputData, const Shape& outputShape) { in transposeConvQuant8PerChannelNhwc()
346 Shape filterChannelShape = filterShape; in transposeConvQuant8PerChannelNhwc()
348 Shape biasChannelShape = biasShape; in transposeConvQuant8PerChannelNhwc()
420 bool transposeConvQuant8PerChannel(const T* inputData, const Shape& inputShape, in transposeConvQuant8PerChannel()
421 const int8_t* filterData, const Shape& filterShape, in transposeConvQuant8PerChannel()
423 const Shape& biasShape, const TransposeConv2dParam& param, in transposeConvQuant8PerChannel()
424 T* outputData, const Shape& outputShape) { in transposeConvQuant8PerChannel()
489 Shape input = context->getInputShape(kInputTensor); in prepare()
490 Shape filter = context->getInputShape(kFilterTensor); in prepare()
491 Shape bias = context->getInputShape(kBiasTensor); in prepare()
536 Shape output = context->getOutputShape(kOutputTensor); in prepare()