HomeSort by relevance Sort by last modified time
    Searched refs:getSizeOfDimension (Results 1 - 9 of 9) sorted by null

  /frameworks/ml/nn/common/
OperationsUtils.cpp 60 uint32_t getSizeOfDimension(const Shape& shape, uint32_t dimensionIdx) {
181 dim1 = getSizeOfDimension(in1, numberOfDims1 - i);
185 dim2 = getSizeOfDimension(in2, numberOfDims2 - i);
232 NN_OPS_CHECK(getSizeOfDimension(filter, 0) == getSizeOfDimension(bias, 0));
233 NN_OPS_CHECK(getSizeOfDimension(filter, 3) == getSizeOfDimension(input, 3));
235 uint32_t channels_out = getSizeOfDimension(filter, 0);
236 uint32_t width = getSizeOfDimension(input, 2);
237 uint32_t height = getSizeOfDimension(input, 1)
    [all...]
CpuExecutor.cpp 424 int32_t input_width = getSizeOfDimension(inputShape, 2);
425 int32_t input_height = getSizeOfDimension(inputShape, 1);
426 int32_t filter_width = getSizeOfDimension(filterShape, 2);
427 int32_t filter_height = getSizeOfDimension(filterShape, 1);
511 int32_t input_width = getSizeOfDimension(inputShape, 2);
512 int32_t input_height = getSizeOfDimension(inputShape, 1);
513 int32_t filter_width = getSizeOfDimension(filterShape, 2);
514 int32_t filter_height = getSizeOfDimension(filterShape, 1);
593 int32_t input_width = getSizeOfDimension(inputShape, 2);
594 int32_t input_height = getSizeOfDimension(inputShape, 1)
    [all...]
  /frameworks/ml/nn/common/operations/
Conv2D.cpp 30 uint32_t height = getSizeOfDimension(inputShape, 1); \
31 uint32_t width = getSizeOfDimension(inputShape, 2); \
32 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
33 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
34 uint32_t outHeight = getSizeOfDimension(outputShape, 1); \
35 uint32_t outWidth = getSizeOfDimension(outputShape, 2); \
36 uint32_t inDepth = getSizeOfDimension(inputShape, 3); \
42 im2colDim.sizes[3] = (int)getSizeOfDimension(outputShape, 0); \
43 im2colDim.sizes[2] = (int)getSizeOfDimension(outputShape, 1); \
44 im2colDim.sizes[1] = (int)getSizeOfDimension(outputShape, 2);
    [all...]
DepthwiseConv2D.cpp 27 uint32_t height = getSizeOfDimension(inputShape, 1); \
28 uint32_t width = getSizeOfDimension(inputShape, 2); \
29 uint32_t filterHeight = getSizeOfDimension(filterShape, 1); \
30 uint32_t filterWidth = getSizeOfDimension(filterShape, 2); \
31 uint32_t outHeight = getSizeOfDimension(outputShape, 1); \
32 uint32_t outWidth = getSizeOfDimension(outputShape, 2); \
Pooling.cpp 26 uint32_t height = getSizeOfDimension(inputShape, 1); \
27 uint32_t width = getSizeOfDimension(inputShape, 2); \
28 uint32_t outHeight = getSizeOfDimension(outputShape, 1); \
29 uint32_t outWidth = getSizeOfDimension(outputShape, 2); \
Reshape.cpp 38 int32_t height = (int32_t) getSizeOfDimension(outputShape, 1);
39 int32_t width = (int32_t) getSizeOfDimension(outputShape, 2);
Activation.cpp 75 uint32_t batch_size = getSizeOfDimension(inputShape, 0);
166 uint32_t batch_size = getSizeOfDimension(inputShape, 0);
  /frameworks/ml/nn/common/operations/internal/
types.h 42 dims.sizes[0] = (int)getSizeOfDimension(shape, 0);
47 dims.sizes[i] = (int)getSizeOfDimension(shape, src);
  /frameworks/ml/nn/common/include/
OperationsUtils.h 68 uint32_t getSizeOfDimension(const Shape& shape, uint32_t dimensionIdx);

Completed in 93 milliseconds