Searched defs:LoadWithCast (Results 1 – 1 of 1) sorted by relevance
104 struct LoadWithCast { struct105 using array_t = at::detail::Array<at::ScalarType, std::max<int>(N, 1)>;106 using size_array_t = at::detail::Array<uint32_t, std::max<int>(N, 1)>;108 array_t dtypes;109 size_array_t element_sizes;111 LoadWithCast(const TensorIteratorBase& iter) { in LoadWithCast() argument121 __device__ scalar_t load(char *base_ptr, uint32_t offset, int arg) { in load()