Searched defs:gather_slice_sizes (Results 1 – 4 of 4) sorted by relevance
213 repeated int64 gather_slice_sizes = 34; field
2110 absl::Span<const int64_t> gather_slice_sizes() const { in gather_slice_sizes() function
843 std::vector<int64_t> gather_slice_sizes; in CreateFromProto() local4822 absl::Span<const int64_t> HloInstruction::gather_slice_sizes() const { in gather_slice_sizes() function in xla::HloInstruction
738 std::vector<int64_t> gather_slice_sizes(output_shape.dimensions().begin(), in RewriteDynamicReshapeCombineInput() local