Lines Matching refs:shape
49 struct Shape {
57 bool SameShape(const Shape& in1, const Shape& in2);
59 // Sets out to the same shape as in.
60 bool SetShape(const Shape& in, Shape* out);
64 uint32_t getNumberOfElements(const Shape& shape);
66 uint32_t getNumberOfDimensions(const Shape& shape);
68 uint32_t getSizeOfDimension(const Shape& shape, uint32_t dimensionIdx);
86 bool GetQuantizedConvolutionMultipler(const Shape& inputShape,
87 const Shape& filterShape,
88 const Shape& biasShape,
89 const Shape& outputShape,
93 const Shape& outputShape,
140 bool addMulPrepare(const Shape& in1, const Shape& in2, Shape* out1);
142 bool floorPrepare(const Shape& input, Shape* output);
144 bool dequantizePrepare(const Shape& input, Shape* output);
146 bool depthwiseConvPrepare(const Shape& input,
147 const Shape& filter,
148 const Shape& bias,
152 Shape* output);
154 bool convPrepare(const Shape& input,
155 const Shape& filter,
156 const Shape& bias,
160 Shape* output);
162 bool genericPoolingPrepare(const Shape& input,
167 Shape* output);
169 bool genericActivationPrepare(const Shape& input, Shape* output);
171 bool fullyConnectedPrepare(const Shape& input,
172 const Shape& weights,
173 const Shape& bias,
174 Shape* output);
176 bool concatenationPrepare(const std::vector<Shape>& inputShapes,
178 Shape* output);
180 bool genericNormalizationPrepare(const Shape& input, Shape* output);
182 bool reshapePrepare(const Shape& input,
185 Shape* output);
187 bool resizeBilinearPrepare(const Shape& input,
190 Shape* output);
192 bool depthToSpacePrepare(const Shape& input,
194 Shape* output);
196 bool spaceToDepthPrepare(const Shape& input,
198 Shape* output);
200 bool embeddingLookupPrepare(const Shape &valueShape,
201 const Shape &lookupShape,
202 Shape *outputShape);
204 bool hashtableLookupPrepare(const Shape &lookupShape,
205 const Shape &keyShape,
206 const Shape &valueShape,
207 Shape *outputShape,
208 Shape *hitShape);
