Searched refs:is_allowed_dim_on_scalar_tensor (Results 1 – 4 of 4) sorted by relevance
71 static bool is_allowed_dim_on_scalar_tensor(int64_t dim) { in is_allowed_dim_on_scalar_tensor() function219 is_allowed_dim_on_scalar_tensor(dim0) && in transpose__batching_rule()220 is_allowed_dim_on_scalar_tensor(dim1)) { in transpose__batching_rule()
348 static bool is_allowed_dim_on_scalar_tensor(int64_t dim) { in is_allowed_dim_on_scalar_tensor() function363 if (/*physical*/self.dim() == 1 && is_allowed_dim_on_scalar_tensor(dim0) && in transpose_int_batch_rule()364 is_allowed_dim_on_scalar_tensor(dim1)) { in transpose_int_batch_rule()
16 static bool is_allowed_dim_on_scalar_tensor(int64_t dim) { in is_allowed_dim_on_scalar_tensor() function171 …bool is_scalar_case = logical_dim == 0 && dims.size() == 1 && is_allowed_dim_on_scalar_tensor(dims… in boxed_reduction_batch_rule()
61 static bool is_allowed_dim_on_scalar_tensor(int64_t dim) { in is_allowed_dim_on_scalar_tensor() function74 …al*/self.dim() == 0 && (dims.empty() || (dims.size() == 1 && is_allowed_dim_on_scalar_tensor(dims[… in sum_batching_rule()333 if (/*logical*/self.dim() == 0 && is_allowed_dim_on_scalar_tensor(dim0) && in transpose_int_batching_rule()334 is_allowed_dim_on_scalar_tensor(dim1)) { in transpose_int_batching_rule()