Searched defs:input_mat (Results 1 – 1 of 1) sorted by relevance
51 static void IndexedShuffle(const int64_t size, const InT& input_mat, in IndexedShuffle()88 const auto input_mat = input.flat_outer_dims<T>(); in RandomShuffle() local