Searched defs:flat_indices (Results 1 – 2 of 2) sorted by relevance
35 int64* __restrict__ flat_indices) { in IndicesFlattenKernel()90 auto flat_indices = flat_indices_tensor.template flat<int64_t>().data(); in operator ()() local
93 auto flat_indices = indices_tensor.flat<int32>(); in FuzzImpl() local