Searched defs:dim_physical (Results 1 – 3 of 3) sorted by relevance
195 auto dim_physical = self_physical.getPhysicalDim(dim); in chunk_batching_rule() local221 auto dim_physical = self_physical.getPhysicalDim(dim); in tensor_split_sections_batching_rule() local229 auto dim_physical = self_physical.getPhysicalDim(dim); in tensor_split_indices_batching_rule() local241 auto dim_physical = in unsqueeze_batching_rule() local296 auto dim_physical = self_physical.getPhysicalDim(dim); in squeeze_dim_batching_rule() local363 auto dim_physical = self_physical.getPhysicalDim(dim); in select_batching_rule() local387 auto dim_physical = self_physical.getPhysicalDim(dim); in slice_batching_rule() local434 auto dim_physical = self_physical.getPhysicalDim(dim); in split_batching_rule() local442 auto dim_physical = self_physical.getPhysicalDim(dim); in split_with_sizes_batching_rule() local450 auto dim_physical = self_physical.getPhysicalDim(dim); in unbind_batching_rule() local[all …]
192 int64_t dim_physical = maybe_wrap_dim(dim, logical_dim + 1); in unsqueeze__batching_rule() local243 auto dim_physical = self_physical.getPhysicalDim(dim); in split_batching_rule() local255 auto dim_physical = self_physical.getPhysicalDim(dim); in split_with_sizes_batching_rule() local267 auto dim_physical = self_physical.getPhysicalDim(dim); in split_with_sizes_copy_batching_rule() local279 auto dim_physical = self_physical.getPhysicalDim(dim); in unbind_batching_rule() local621 auto dim_physical = in stack_batching_rule() local
266 auto dim_physical = getPhysicalDim(_self, true, dim); in select_batching_rule() local