Searched defs:output_flat_inds (Results 1 – 1 of 1) sorted by relevance
48 int64* __restrict__ output_flat_inds) { in SparseConcatKernel()88 const int64* __restrict__ output_flat_inds, in SparseConcatPermuteKernel()180 Tensor output_flat_inds; in operator ()() local