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