Lines Matching refs:outputShape
58 _Float16* outputData, const Shape& outputShape);
65 const Shape& outputShape);
72 uint8_t* outputData, const Shape& outputShape);
81 const Shape& outputShape);
85 _Float16* outputData, const Shape& outputShape);
88 const Shape& outputShape);
91 const Shape& outputShape);
95 T* outputData, const Shape& outputShape);
98 T* outputData, const Shape& outputShape);
102 T* outputData, const Shape& outputShape);
106 T* outputData, const Shape& outputShape);
111 const Shape& outputShape);
115 const Shape& outputShape);
118 bool keepDims, T* outputData, const Shape& outputShape);
123 int32_t shrinkAxisMask, uint8_t* outputData, const Shape& outputShape);
126 bool isArgMin, uint8_t* outputData, const Shape& outputShape);
153 int32_t activation, _Float16* outputData, const Shape& outputShape);
160 const Shape& outputShape);
168 const Shape& outputShape);
177 int32_t activation, T* outputData, const Shape& outputShape);
180 int32_t axis, uint8_t* outputData, const Shape& outputShape);