Searched defs:gather_slice_sizes (Results 1 – 4 of 4) sorted by relevance
193 repeated int64 gather_slice_sizes = 34; field
1764 absl::Span<const int64> gather_slice_sizes() const { in gather_slice_sizes() function
683 std::vector<int64> gather_slice_sizes(output_shape.dimensions().begin(), in RewriteDynamicReshapeCombineInput() local
785 std::vector<int64> gather_slice_sizes; in CreateFromProto() local4544 absl::Span<const int64> HloInstruction::gather_slice_sizes() const { in gather_slice_sizes() function in xla::HloInstruction