Searched defs:GatherParallelDims (Results 1 – 1 of 1) sorted by relevance
34 struct GatherParallelDims { struct35 absl::InlinedVector<int64_t, 1> indices_parallel_dims;36 absl::InlinedVector<int64_t, 1> operand_parallel_dims;37 std::vector<int64_t> index_parallel_in_dim;