Searched defs:compute_strides (Results 1 – 2 of 2) sorted by relevance
41 inline Strides compute_strides(const ITensorInfo &info, T stride_x, Ts &&... fixed_strides) in compute_strides() function63 inline Strides compute_strides(const ITensorInfo &info) in compute_strides() function
1270 void TensorIteratorBase::compute_strides(const TensorIteratorConfig& config) { in compute_strides() function in at::TensorIteratorBase