Searched defs:upper_dims (Results 1 – 8 of 8) sorted by relevance
200 const unsigned int upper_dims = src.shape().total_size_upper(1); in compute_reduction_operation() local212 const unsigned int upper_dims = src.shape().total_size_upper(2); in compute_reduction_operation() local229 const unsigned int upper_dims = src.shape().total_size_upper(3); in compute_reduction_operation() local249 const unsigned int upper_dims = src.shape().total_size_upper(4); in compute_reduction_operation() local
48 const int upper_dims = src.shape().total_size() / (cols * rows * depth); in batch_normalization_layer() local
60 const int upper_dims = src.shape().total_size_upper(axis + 1); in l2_normalize() local
45 const int upper_dims = src.shape().total_size() / (cols * rows * depth); in normalize_planar_yuv_layer() local
51 int upper_dims = src.shape().total_size() / (cols * rows); in normalization_layer() local
61 const int upper_dims = src.shape().total_size() / (width * height * depth); in widthconcatenate_layer() local
195 const unsigned int upper_dims = src.shape().total_size_upper(1); in rdft_1d_core() local218 const unsigned int upper_dims = src.shape().total_size_upper(1); in dft_1d_core() local
66 const int upper_dims = src.shape().total_size() / (w_src * h_src); in pooling_layer_internal() local