Searched refs:output_gather_index (Results 1 – 1 of 1) sorted by relevance
1088 absl::Span<const int64> output_gather_index) -> StatusOr<bool> { in HandleGather() argument1093 output_index[i] = output_gather_index[i] + output_window_index[i]; in HandleGather()1123 [&](absl::Span<const int64> output_gather_index) -> StatusOr<bool> { in HandleGather() argument1125 output_batch_index_to_input_index(output_gather_index)); in HandleGather()1129 input_gather_index, output_gather_index))); in HandleGather()