OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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