Operations.h - OpenGrok cross reference for /frameworks/ml/nn/common/include/Operations.h

Lines Matching refs:inputShape
49 bool depthwiseConvFloat16(const _Float16* inputData, const Shape& inputShape,
56 bool depthwiseConvFloat32(const float* inputData, const Shape& inputShape, const float* filterData,
63 bool depthwiseConvQuant8(const uint8_t* inputData, const Shape& inputShape,
70 bool depthwiseConvQuant8PerChannel(const uint8_t* inputData, const Shape& inputShape,
80 bool localResponseNormFloat16(const _Float16* inputData, const Shape& inputShape, int32_t radius,
83 bool localResponseNormFloat32(const float* inputData, const Shape& inputShape, int32_t radius,
87 bool copyData(const void* inputData, const Shape& inputShape, void* outputData,
91 bool depthToSpaceGeneric(const T* inputData, const Shape& inputShape, int32_t blockSize,
94 bool spaceToDepthGeneric(const T* inputData, const Shape& inputShape, int32_t blockSize,
98 bool padGeneric(const T* inputData, const Shape& inputShape, const int32_t* paddings, T pad_value,
102 bool batchToSpaceGeneric(const T* inputData, const Shape& inputShape, const int32_t* blockSize,
106 bool spaceToBatchGeneric(const T* inputData, const Shape& inputShape, const int32_t* blockSize,
110 bool meanFloat16(_Float16* inputData, const Shape& inputShape, const int32_t* axis,
114 bool meanGeneric(T* inputData, const Shape& inputShape, const int32_t* axis, const Shape& axisShape,
117 bool stridedSliceGeneric(const uint8_t* inputData, const Shape& inputShape,
122 bool argMinMaxGeneric(const uint8_t* inputData, const Shape& inputShape, int32_t axis,
125 bool splitFloat16(const _Float16* inputData, const Shape& inputShape, int32_t axis,
129 bool splitFloat32(const float* inputData, const Shape& inputShape, const int32_t axis,
133 bool splitInt32(const int32_t* inputData, const Shape& inputShape, const int32_t axis,
137 bool splitQuant8(const uint8_t* inputData, const Shape& inputShape, const int32_t axis,
141 bool groupedConvFloat16(const _Float16* inputData, const Shape& inputShape,
148 bool groupedConvFloat32(const float* inputData, const Shape& inputShape, const float* filterData,
155 bool groupedConvQuant8(const uint8_t* inputData, const Shape& inputShape, const uint8_t* filterData,
162 bool groupedConvQuant8PerChannel(const uint8_t* inputData, const Shape& inputShape,
170 bool channelShuffleGeneric(const uint8_t* inputData, const Shape& inputShape, int32_t numGroups,