Searched defs:vec_mean (Results 1 – 2 of 2) sorted by relevance
48 inline float16x8_t vector_float_norm(const float16x8_t &inputs, const float32x4_t &vec_mean, const … in vector_float_norm()61 InputType vector_float_norm(const InputType &inputs, const AccType &vec_mean, const AccType &vec_mu… in vector_float_norm()
428 const GenericPackedTensorAccessor<accscalar_t, 2, RestrictPtrTraits, index_t> vec_mean, in batch_norm_reduce_statistics_kernel()