Searched defs:normalized_ndim (Results 1 – 7 of 7) sorted by relevance
17 const int normalized_ndim = normalized_shape.size(); in _check_layer_norm_inputs() local
222 const int normalized_ndim = normalized_shape.size(); in math_native_layer_norm() local
34 const size_t normalized_ndim = normalized_shape.size(); in _check_nested_layer_norm_inputs() local
17 const auto normalized_ndim = normalized_shape.size(); in _check_layer_norm_inputs() local
75 const int64_t normalized_ndim = pop(stack).toInt(); in __anon6bc4869d0302() local
477 const auto normalized_ndim = normalized_shape.size(); in _check_layer_norm_inputs() local
4800 const auto normalized_ndim = normalized_shape.size(); in layer_norm_double_backward() local