Searched refs:get_input_index (Results 1 – 1 of 1) sorted by relevance
153 auto get_input_index = [is_transpose, &indices_mat](int64 i) { in Compute() local176 [&get_input_index](int64 i, int64 j) { in Compute()177 return get_input_index(i) < get_input_index(j); in Compute()188 get_input_index(perm[start]) == get_input_index(perm[end])) { in Compute()242 const int64 input_index = get_input_index(perm[shard.first]); in Compute()256 CHECK_EQ(input_index, get_input_index(i)); in Compute()