Searched refs:checkBatchDimsAtFrontInLayout (Results 1 – 1 of 1) sorted by relevance
491 static void checkBatchDimsAtFrontInLayout(IntArrayRef physical_strides, int64_t num_batch_dims) { in checkBatchDimsAtFrontInLayout() function577 checkBatchDimsAtFrontInLayout(self_physical_tensor.strides(), num_batch_dims); in _new_zeros_with_same_feature_meta_batching_rule()635 checkBatchDimsAtFrontInLayout(physical_tensor.strides(), num_batch_dims); in as_strided_batching_rule()