Searched defs:get_strides (Results 1 – 3 of 3) sorted by relevance
46 inline void get_strides(int64_t* strides, ArrayRef<OperandInfo> operands, int64_t ndim) { in get_strides() function775 StrideVector TensorIteratorBase::get_strides() const { in get_strides() function in at::TensorIteratorBase
55 …auto get_strides = [](int64_t* strides, IntArrayRef strides_out, IntArrayRef strides_in, int64_t n… in reduced_float_copy_kernel() local
568 const auto get_strides = [](const Shape& shape) { in ShapeIsCompatible() local