Home
last modified time | relevance | path

Searched defs:slice_numel (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/csrc/jit/runtime/static/
Dops.cpp147 auto slice_numel = std::accumulate( in flatten_copy_out() local
/external/pytorch/aten/src/ATen/native/
DTensorShape.cpp3453 …auto slice_numel = c10::multiply_integers(self.sym_sizes().slice(start_dim, end_dim - start_dim + … in flatten() local