Searched defs:scatter_dims_to_operand_dims (Results 1 – 10 of 10) sorted by relevance
69 std::vector<int64_t> scatter_dims_to_operand_dims( in ConvertScatterDimensionNumbers() local
355 auto scatter_dims_to_operand_dims = in Convert_scatter_dimension_numbers() local
541 std::vector<int64> scatter_dims_to_operand_dims( in HandleScatter() local
1055 std::vector<int64> scatter_dims_to_operand_dims( in ScatterOutputShardingFromUpdate() local1079 std::vector<int64> scatter_dims_to_operand_dims( in ScatterUpdateShardingFromOutput() local
3005 string scatter_dims_to_operand_dims = StrCat( in ScatterDimensionNumbersToString() local3022 absl::Span<const int64> scatter_dims_to_operand_dims, in MakeScatterDimNumbers()
2648 optional<std::vector<int64>> scatter_dims_to_operand_dims; in ParseInstructionRhs() local
516 repeated int64 scatter_dims_to_operand_dims = 3; field
1795 auto scatter_dims_to_operand_dims = in matchAndRewrite() local
2634 auto scatter_dims_to_operand_dims = Extract1DVector( in matchAndRewrite() local
6017 SmallVector<int64_t, 1> scatter_dims_to_operand_dims(1, 0); in matchAndRewrite() local