Home
last modified time | relevance | path

Searched refs:DeviceDtypeKey (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/
DForeachUtils.h275 using DeviceDtypeKey = std::pair<at::Device, at::ScalarType>; variable
281 DeviceDtypeKey,
283 ParamsHash<DeviceDtypeKey>>;
306 const auto key = [&]() -> DeviceDtypeKey { in _group_tensors_by_first_tensors_device_and_dtype()