Searched refs:StoreWithCastVariant (Results 1 – 2 of 2) sorted by relevance
217 struct StoreWithCastVariant { struct224 StoreWithCastVariant(const TensorIteratorBase& iter) { in StoreWithCastVariant() argument
244 StoreWithCastVariant storer(iter); in jitted_gpu_kernel_dynamic_impl()