Home
last modified time | relevance | path

Searched defs:input_depth (Results 1 – 3 of 3) sorted by relevance

/frameworks/ml/nn/common/operations/internal/reference/
Ddepthwiseconv_float.h38 const int input_depth = ArraySize(input_dims, 0); in DepthwiseConv() local
Ddepthwiseconv_uint8.h54 const int input_depth = ArraySize(input_dims, 0); in DepthwiseConv() local
Dreference_ops.h169 const int input_depth = MatchingArraySize(input_dims, 0, filter_dims, 0); in Conv() local
242 const int input_depth = MatchingArraySize(input_dims, 0, filter_dims, 0); in Conv() local
298 const int input_depth = ArraySize(input_dims, 0); in DepthToSpace() local
339 const int input_depth = ArraySize(input_dims, 0); in SpaceToDepth() local
1071 const int input_depth = ArraySize(input_dims, 0); in LstmCell() local