Searched defs:StoreWithCast (Results 1 – 1 of 1) sorted by relevance
135 struct StoreWithCast { struct136 using array_t = at::detail::Array<at::ScalarType, std::max<int>(N, 1)>;137 using size_array_t = at::detail::Array<uint32_t, std::max<int>(N, 1)>;139 array_t dtypes;140 size_array_t element_sizes;142 StoreWithCast(const TensorIteratorBase& iter) { in StoreWithCast() argument152 __device__ void store(scalar_t value, char *base_ptr, uint32_t offset, int arg = 0) { in store()