Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/sparse/
DSparseTensor.cpp422 auto cpu_max_indices_accessor = cpu_max_indices.accessor<int64_t, 1>(); in _validate_sparse_coo_tensor_args() local
433 int64_t max_index_in_dim = cpu_max_indices_accessor[d]; in _validate_sparse_coo_tensor_args()