Searched defs:inserted_window_dims (Results 1 – 9 of 9) sorted by relevance
63 std::vector<int64_t> inserted_window_dims( in ConvertScatterDimensionNumbers() local
441 auto inserted_window_dims = input.getInsertedWindowDims(); in Convert_scatter_dimension_numbers() local
41 absl::Span<const int64_t> inserted_window_dims, in FlattenAndTransposeUpdates()
1136 std::vector<int64_t> inserted_window_dims( in ScatterOutputShardingFromUpdate() local1155 std::vector<int64_t> inserted_window_dims( in ScatterUpdateShardingFromOutput() local
3162 std::string inserted_window_dims = StrCat( in ScatterDimensionNumbersToString() local3181 absl::Span<const int64_t> inserted_window_dims, in MakeScatterDimNumbers()
2850 optional<std::vector<int64_t>> inserted_window_dims; in CreateInstruction() local
556 repeated int64 inserted_window_dims = 2; field
3175 auto inserted_window_dims = in matchAndRewrite() local
5738 SmallVector<int64_t, 1> inserted_window_dims(1, 0); in matchAndRewrite() local