Searched defs:GatherParallelDims (Results 1 – 1 of 1) sorted by relevance
33 struct GatherParallelDims { struct34 absl::InlinedVector<int64, 1> indices_parallel_dims;35 absl::InlinedVector<int64, 1> operand_parallel_dims;36 std::vector<int64> index_parallel_in_dim;