Searched defs:elements_per_channel (Results 1 – 2 of 2) sorted by relevance
242 size_t elements_per_channel = N * inner; in _native_batch_norm_legit_no_stats_out() local
3736 int64_t elements_per_channel = size_from_dim_(axis + 1, rtensor.sizes()); local3788 int64_t elements_per_channel = size_from_dim_(axis + 1, rtensor.sizes()); local4003 int64_t elements_per_channel = local4080 int64_t elements_per_channel = local