• Home
  • Raw
  • Download

Lines Matching refs:Shape

28 bool SameShape(const Shape& in1, const Shape& in2) {  in SameShape()
40 bool SetShape(const Shape& in, Shape* out) { in SetShape()
48 uint32_t getNumberOfElements(const Shape& shape) { in getNumberOfElements()
56 uint32_t getNumberOfDimensions(const Shape& shape) { in getNumberOfDimensions()
60 uint32_t getSizeOfDimension(const Shape& shape, uint32_t dimensionIdx) { in getSizeOfDimension()
110 bool GetQuantizedConvolutionMultipler(const Shape& inputShape, in GetQuantizedConvolutionMultipler()
111 const Shape& filterShape, in GetQuantizedConvolutionMultipler()
112 const Shape& biasShape, in GetQuantizedConvolutionMultipler()
113 const Shape& outputShape, in GetQuantizedConvolutionMultipler()
129 const Shape& outputShape, in CalculateActivationRangeUint8()
167 bool addMulPrepare(const Shape& in1, const Shape& in2, Shape* out) { in addMulPrepare()
197 bool floorPrepare(const Shape& input, Shape* output) { in floorPrepare()
201 bool dequantizePrepare(const Shape& input, Shape* output) { in dequantizePrepare()
215 bool convPrepare(const Shape& input, in convPrepare()
216 const Shape& filter, in convPrepare()
217 const Shape& bias, in convPrepare()
221 Shape* output) { in convPrepare()
252 bool depthwiseConvPrepare(const Shape& input, in depthwiseConvPrepare()
253 const Shape& filter, in depthwiseConvPrepare()
254 const Shape& bias, in depthwiseConvPrepare()
258 Shape* output) { in depthwiseConvPrepare()
289 bool genericPoolingPrepare(const Shape& input, in genericPoolingPrepare()
294 Shape* output) { in genericPoolingPrepare()
313 bool genericActivationPrepare(const Shape& input, in genericActivationPrepare()
314 Shape* output) { in genericActivationPrepare()
319 bool fullyConnectedPrepare(const Shape& input, in fullyConnectedPrepare()
320 const Shape& weights, in fullyConnectedPrepare()
321 const Shape& bias, in fullyConnectedPrepare()
322 Shape* output) { in fullyConnectedPrepare()
346 bool concatenationPrepare(const std::vector<Shape>& inputShapes, in concatenationPrepare()
348 Shape* output) { in concatenationPrepare()
388 bool genericNormalizationPrepare(const Shape& input, Shape* output) { in genericNormalizationPrepare()
393 bool reshapePrepare(const Shape& input, in reshapePrepare()
396 Shape* output) { in reshapePrepare()
432 bool resizeBilinearPrepare(const Shape& input, in resizeBilinearPrepare()
435 Shape* output) { in resizeBilinearPrepare()
446 bool depthToSpacePrepare(const Shape& input, in depthToSpacePrepare()
448 Shape* output) { in depthToSpacePrepare()
469 bool spaceToDepthPrepare(const Shape& input, in spaceToDepthPrepare()
471 Shape* output) { in spaceToDepthPrepare()
494 bool embeddingLookupPrepare(const Shape &valueShape, in embeddingLookupPrepare()
495 const Shape &lookupShape, in embeddingLookupPrepare()
496 Shape *outputShape) { in embeddingLookupPrepare()
516 bool hashtableLookupPrepare(const Shape &lookupShape, in hashtableLookupPrepare()
517 const Shape &keyShape, in hashtableLookupPrepare()
518 const Shape &valueShape, in hashtableLookupPrepare()
519 Shape *outputShape, in hashtableLookupPrepare()
520 Shape *hitShape) { in hashtableLookupPrepare()