Cross Reference: /frameworks/ml/nn/common/operations/Reshape.cpp

Lines Matching refs:inputShape
29 bool reshapeGeneric(const void* inputData, const Shape& inputShape,
31     size_t count = sizeOfData(inputShape.type, inputShape.dimensions);
36 bool resizeBilinearFloat32(const float* inputData, const Shape& inputShape,
47             inputData, convertShapeToDims(inputShape),
53 bool depthToSpaceGeneric(const uint8_t* inputData, const Shape& inputShape,
56     if (inputShape.type == OperandType::TENSOR_FLOAT32) {
59                 convertShapeToDims(inputShape),
63     } else if (inputShape.type == OperandType::TENSOR_QUANT8_ASYMM) {
66                 convertShapeToDims(inputShape),
77 bool spaceToDepthGeneric(const uint8_t* inputData, const Shape& inputShape,
80     if (inputShape.type == OperandType::TENSOR_FLOAT32) {
83                 convertShapeToDims(inputShape),
87     } else if (inputShape.type == OperandType::TENSOR_QUANT8_ASYMM) {
90                 convertShapeToDims(inputShape),