Searched defs:slice_dim_size (Results 1 – 2 of 2) sorted by relevance
2676 const int64_t slice_dim_size = slice_sizes[dim]; in InferDynamicSliceShape() local
615 auto slice_dim_size = source_slice.sizes()[dim]; in TORCH_IMPL_FUNC() local644 auto slice_dim_size = tensor.sizes()[dim]; in TORCH_IMPL_FUNC() local