Searched defs:inserted_window_dims (Results 1 – 8 of 8) sorted by relevance
67 std::vector<int64_t> inserted_window_dims( in ConvertScatterDimensionNumbers() local
350 auto inserted_window_dims = ConvertDenseIntAttr(input.inserted_window_dims()); in Convert_scatter_dimension_numbers() local
1053 std::vector<int64> inserted_window_dims(dnums.inserted_window_dims().begin(), in ScatterOutputShardingFromUpdate() local1077 std::vector<int64> inserted_window_dims(dnums.inserted_window_dims().begin(), in ScatterUpdateShardingFromOutput() local
3002 string inserted_window_dims = StrCat( in ScatterDimensionNumbersToString() local3021 absl::Span<const int64> inserted_window_dims, in MakeScatterDimNumbers()
2645 optional<std::vector<int64>> inserted_window_dims; in ParseInstructionRhs() local
514 repeated int64 inserted_window_dims = 2; field
1793 auto inserted_window_dims = in matchAndRewrite() local
6016 SmallVector<int64_t, 1> inserted_window_dims(1, 0); in matchAndRewrite() local