Searched defs:input_mat (Results 1 – 2 of 2) sorted by relevance
50 static void IndexedShuffle(const int64 size, const InT& input_mat, in IndexedShuffle()93 const auto input_mat = input.flat_outer_dims<T>(); in Compute() local
98 def calculate_loss(input_mat, row_factors, col_factors, regularization=None, argument