Searched defs:indices_sizes (Results 1 – 5 of 5) sorted by relevance
124 Eigen::DSizes<Eigen::DenseIndex, 2> indices_sizes(num_entries, rank); in Compute() local
446 Eigen::DSizes<Eigen::DenseIndex, 2> indices_sizes(num_entries, rank); in Compute() local
284 const auto indices_sizes = indices.sizes(); in embedding_bag_nbit_impl() local454 const auto indices_sizes = indices.sizes(); in embedding_bag_byte_impl() local
645 DimVector indices_sizes(indices.sizes().begin(), indices.sizes().end()); in max_pool3d_kernel_impl() local
2175 SymDimVector indices_sizes{indices.sym_sizes()}; in _take_along_dim_helper() local