Searched defs:input_ndim (Results 1 – 8 of 8) sorted by relevance
72 const auto input_ndim = qx_contig.dim(); in quantized_group_norm_impl() local118 const auto input_ndim = qx.dim(); in quantized_instance_norm_impl() local
39 const auto input_ndim = input.dim(); in _check_layer_norm_inputs() local
221 const auto input_ndim = input.dim(); in math_native_layer_norm() local
39 const auto input_ndim = input.sizes().size(); in _check_layer_norm_inputs() local
78 const int64_t input_ndim = input_shape.size(); in __anon90d7ec1d0302() local
42 const auto input_ndim = self.dim(); in cpu_cum_base_kernel() local
679 int64_t input_ndim = qx.dim(); in qprelu_out_kernel() local
4802 const auto input_ndim = input_t.dim(); in layer_norm_double_backward() local