Searched refs:DeviceDtypeKey (Results 1 – 1 of 1) sorted by relevance
275 using DeviceDtypeKey = std::pair<at::Device, at::ScalarType>; variable281 DeviceDtypeKey,283 ParamsHash<DeviceDtypeKey>>;306 const auto key = [&]() -> DeviceDtypeKey { in _group_tensors_by_first_tensors_device_and_dtype()