Searched defs:update_window_dims (Results 1 – 9 of 9) sorted by relevance
61 std::vector<int64_t> update_window_dims(dnums.update_window_dims().begin(), in ConvertScatterDimensionNumbers() local
436 auto update_window_dims = input.getUpdateWindowDims(); in Convert_scatter_dimension_numbers() local
40 HloInstruction* updates, absl::Span<const int64_t> update_window_dims, in FlattenAndTransposeUpdates()
98 HloInstruction* updates, absl::Span<const int64_t> update_window_dims) { in PermuteScatterAndWindowDims()
1141 std::vector<int64_t> update_window_dims(dnums.update_window_dims().begin(), in ScatterOutputShardingFromUpdate() local1160 std::vector<int64_t> update_window_dims(dnums.update_window_dims().begin(), in ScatterUpdateShardingFromOutput() local
3159 std::string update_window_dims = in ScatterDimensionNumbersToString() local3180 absl::Span<const int64_t> update_window_dims, in MakeScatterDimNumbers()
2847 optional<std::vector<int64_t>> update_window_dims; in CreateInstruction() local
554 repeated int64 update_window_dims = 1; field
2799 Operation *scatter_op, llvm::ArrayRef<int64_t> update_window_dims, in CanonicalizeScatterUpdates()3168 auto update_window_dims = scatter_dimension_numbers.getUpdateWindowDims(); in matchAndRewrite() local