Searched defs:update_window_dims (Results 1 – 9 of 9) sorted by relevance
65 std::vector<int64_t> update_window_dims(dnums.update_window_dims().begin(), in ConvertScatterDimensionNumbers() local
345 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()
1058 std::vector<int64> update_window_dims(dnums.update_window_dims().begin(), in ScatterOutputShardingFromUpdate() local1082 std::vector<int64> update_window_dims(dnums.update_window_dims().begin(), in ScatterUpdateShardingFromOutput() local
2999 string update_window_dims = in ScatterDimensionNumbersToString() local3020 absl::Span<const int64> update_window_dims, in MakeScatterDimNumbers()
2642 optional<std::vector<int64>> update_window_dims; in ParseInstructionRhs() local
512 repeated int64 update_window_dims = 1; field
1810 auto update_window_dims = in matchAndRewrite() local
2623 auto update_window_dims = in matchAndRewrite() local