Searched defs:update_window_dims (Results 1 – 8 of 8) sorted by relevance
65 std::vector<int64_t> update_window_dims(dnums.update_window_dims().begin(), in ConvertScatterDimensionNumbers() local
321 auto update_window_dims = ConvertDenseIntAttr(input.update_window_dims()); in Convert_scatter_dimension_numbers() local
95 HloInstruction* updates, absl::Span<const int64> update_window_dims) { in PermuteScatterAndWindowDims()
1015 std::vector<int64> update_window_dims(dnums.update_window_dims().begin(), in ScatterOutputShardingFromUpdate() local1039 std::vector<int64> update_window_dims(dnums.update_window_dims().begin(), in ScatterUpdateShardingFromOutput() local
2835 string update_window_dims = in ScatterDimensionNumbersToString() local2856 absl::Span<const int64> update_window_dims, in MakeScatterDimNumbers()
2486 optional<std::vector<int64>> update_window_dims; in ParseInstructionRhs() local
506 repeated int64 update_window_dims = 1; field
1135 auto update_window_dims = in matchAndRewrite() local