@@ -156,12 +156,12 @@ structured_op: !LinalgStructuredOpConfig
156
156
assignments :
157
157
- !ScalarAssign
158
158
arg : O
159
- value : !Scalarlogression
159
+ value : !ScalarExpression
160
160
scalar_fn :
161
161
kind : unary
162
162
fn_name : log
163
163
operands :
164
- - !Scalarlogression
164
+ - !ScalarExpression
165
165
scalar_arg : I
166
166
--- !LinalgOpConfig
167
167
metadata : !LinalgOpMetadata
@@ -191,7 +191,7 @@ structured_op: !LinalgStructuredOpConfig
191
191
assignments :
192
192
- !ScalarAssign
193
193
arg : O
194
- value : !Scalarabsression
194
+ value : !ScalarExpression
195
195
scalar_fn :
196
196
kind : unary
197
197
fn_name : abs
@@ -226,12 +226,12 @@ structured_op: !LinalgStructuredOpConfig
226
226
assignments :
227
227
- !ScalarAssign
228
228
arg : O
229
- value : !Scalarceilression
229
+ value : !ScalarExpression
230
230
scalar_fn :
231
231
kind : unary
232
232
fn_name : ceil
233
233
operands :
234
- - !Scalarceilression
234
+ - !ScalarExpression
235
235
scalar_arg : I
236
236
--- !LinalgOpConfig
237
237
metadata : !LinalgOpMetadata
@@ -261,17 +261,17 @@ structured_op: !LinalgStructuredOpConfig
261
261
assignments :
262
262
- !ScalarAssign
263
263
arg : O
264
- value : !Scalarfloorression
264
+ value : !ScalarExpression
265
265
scalar_fn :
266
266
kind : unary
267
267
fn_name : floor
268
268
operands :
269
- - !Scalarfloorression
269
+ - !ScalarExpression
270
270
scalar_arg : I
271
271
--- !LinalgOpConfig
272
272
metadata : !LinalgOpMetadata
273
273
name : negf
274
- cpp_class_name : NegFOp
274
+ cpp_class_name : NegfOp
275
275
doc : |-
276
276
Applies negf(x) elementwise.
277
277
@@ -296,7 +296,7 @@ structured_op: !LinalgStructuredOpConfig
296
296
assignments :
297
297
- !ScalarAssign
298
298
arg : O
299
- value : !Scalarnegfression
299
+ value : !ScalarExpression
300
300
scalar_fn :
301
301
kind : unary
302
302
fn_name : negf
@@ -386,17 +386,17 @@ structured_op: !LinalgStructuredOpConfig
386
386
- !LinalgOperandDefConfig
387
387
name : lhs
388
388
kind : input_tensor
389
- type_var : T
389
+ type_var : T1
390
390
shape_map : affine_map<() -> ()>
391
391
- !LinalgOperandDefConfig
392
392
name : rhs
393
393
kind : input_tensor
394
- type_var : T
394
+ type_var : T1
395
395
shape_map : affine_map<() -> ()>
396
396
- !LinalgOperandDefConfig
397
- name : out
397
+ name : O
398
398
kind : output_tensor
399
- type_var : T
399
+ type_var : T1
400
400
shape_map : affine_map<() -> ()>
401
401
indexing_maps : !LinalgIndexingMapsConfig
402
402
static_indexing_maps :
@@ -406,7 +406,7 @@ structured_op: !LinalgStructuredOpConfig
406
406
iterator_types : []
407
407
assignments :
408
408
- !ScalarAssign
409
- arg : out
409
+ arg : O
410
410
value : !ScalarExpression
411
411
scalar_fn :
412
412
kind : binary
@@ -435,17 +435,17 @@ structured_op: !LinalgStructuredOpConfig
435
435
- !LinalgOperandDefConfig
436
436
name : lhs
437
437
kind : input_tensor
438
- type_var : T
438
+ type_var : T1
439
439
shape_map : affine_map<() -> ()>
440
440
- !LinalgOperandDefConfig
441
441
name : rhs
442
442
kind : input_tensor
443
- type_var : T
443
+ type_var : T1
444
444
shape_map : affine_map<() -> ()>
445
445
- !LinalgOperandDefConfig
446
- name : out
446
+ name : O
447
447
kind : output_tensor
448
- type_var : T
448
+ type_var : T1
449
449
shape_map : affine_map<() -> ()>
450
450
indexing_maps : !LinalgIndexingMapsConfig
451
451
static_indexing_maps :
@@ -455,7 +455,7 @@ structured_op: !LinalgStructuredOpConfig
455
455
iterator_types : []
456
456
assignments :
457
457
- !ScalarAssign
458
- arg : out
458
+ arg : O
459
459
value : !ScalarExpression
460
460
scalar_fn :
461
461
kind : binary
@@ -470,7 +470,7 @@ metadata: !LinalgOpMetadata
470
470
name : mul
471
471
cpp_class_name : MulOp
472
472
doc : |-
473
- Multiply two tensors elementwise.
473
+ Multiplies two tensors elementwise.
474
474
475
475
The shapes and element types must be identical. The appropriate casts,
476
476
broadcasts and reductions should be done previously to calling this op.
@@ -484,17 +484,17 @@ structured_op: !LinalgStructuredOpConfig
484
484
- !LinalgOperandDefConfig
485
485
name : lhs
486
486
kind : input_tensor
487
- type_var : T
487
+ type_var : T1
488
488
shape_map : affine_map<() -> ()>
489
489
- !LinalgOperandDefConfig
490
490
name : rhs
491
491
kind : input_tensor
492
- type_var : T
492
+ type_var : T1
493
493
shape_map : affine_map<() -> ()>
494
494
- !LinalgOperandDefConfig
495
- name : out
495
+ name : O
496
496
kind : output_tensor
497
- type_var : T
497
+ type_var : T1
498
498
shape_map : affine_map<() -> ()>
499
499
indexing_maps : !LinalgIndexingMapsConfig
500
500
static_indexing_maps :
@@ -504,7 +504,7 @@ structured_op: !LinalgStructuredOpConfig
504
504
iterator_types : []
505
505
assignments :
506
506
- !ScalarAssign
507
- arg : out
507
+ arg : O
508
508
value : !ScalarExpression
509
509
scalar_fn :
510
510
kind : binary
@@ -519,8 +519,7 @@ metadata: !LinalgOpMetadata
519
519
name : div
520
520
cpp_class_name : DivOp
521
521
doc : |-
522
- Divides the first tensor by the second tensor, elementwise. For integer
523
- types, performs a signed division.
522
+ Divides the first tensor by the second tensor, elementwise.
524
523
525
524
The shapes and element types must be identical. The appropriate casts,
526
525
broadcasts and reductions should be done previously to calling this op.
@@ -534,17 +533,17 @@ structured_op: !LinalgStructuredOpConfig
534
533
- !LinalgOperandDefConfig
535
534
name : lhs
536
535
kind : input_tensor
537
- type_var : T
536
+ type_var : T1
538
537
shape_map : affine_map<() -> ()>
539
538
- !LinalgOperandDefConfig
540
539
name : rhs
541
540
kind : input_tensor
542
- type_var : T
541
+ type_var : T1
543
542
shape_map : affine_map<() -> ()>
544
543
- !LinalgOperandDefConfig
545
- name : out
544
+ name : O
546
545
kind : output_tensor
547
- type_var : T
546
+ type_var : T1
548
547
shape_map : affine_map<() -> ()>
549
548
indexing_maps : !LinalgIndexingMapsConfig
550
549
static_indexing_maps :
@@ -554,7 +553,7 @@ structured_op: !LinalgStructuredOpConfig
554
553
iterator_types : []
555
554
assignments :
556
555
- !ScalarAssign
557
- arg : out
556
+ arg : O
558
557
value : !ScalarExpression
559
558
scalar_fn :
560
559
kind : binary
@@ -567,7 +566,7 @@ structured_op: !LinalgStructuredOpConfig
567
566
--- !LinalgOpConfig
568
567
metadata : !LinalgOpMetadata
569
568
name : div_unsigned
570
- cpp_class_name : DivUOp
569
+ cpp_class_name : DivUnsignedOp
571
570
doc : |-
572
571
Divides the first tensor by the second tensor, elementwise. For integer
573
572
types, performs an unsigned division.
@@ -584,17 +583,17 @@ structured_op: !LinalgStructuredOpConfig
584
583
- !LinalgOperandDefConfig
585
584
name : lhs
586
585
kind : input_tensor
587
- type_var : T
586
+ type_var : T1
588
587
shape_map : affine_map<() -> ()>
589
588
- !LinalgOperandDefConfig
590
589
name : rhs
591
590
kind : input_tensor
592
- type_var : T
591
+ type_var : T1
593
592
shape_map : affine_map<() -> ()>
594
593
- !LinalgOperandDefConfig
595
- name : out
594
+ name : O
596
595
kind : output_tensor
597
- type_var : T
596
+ type_var : T1
598
597
shape_map : affine_map<() -> ()>
599
598
indexing_maps : !LinalgIndexingMapsConfig
600
599
static_indexing_maps :
@@ -604,7 +603,7 @@ structured_op: !LinalgStructuredOpConfig
604
603
iterator_types : []
605
604
assignments :
606
605
- !ScalarAssign
607
- arg : out
606
+ arg : O
608
607
value : !ScalarExpression
609
608
scalar_fn :
610
609
kind : binary
@@ -619,7 +618,7 @@ metadata: !LinalgOpMetadata
619
618
name : max
620
619
cpp_class_name : MaxOp
621
620
doc : |-
622
- Takes the max (signed) between the input and a constant .
621
+ Takes the max (signed) between two inputs, elementwise .
623
622
624
623
The shapes and element types must be identical. The appropriate casts,
625
624
broadcasts and reductions should be done previously to calling this op.
@@ -633,17 +632,17 @@ structured_op: !LinalgStructuredOpConfig
633
632
- !LinalgOperandDefConfig
634
633
name : lhs
635
634
kind : input_tensor
636
- type_var : T
635
+ type_var : T1
637
636
shape_map : affine_map<() -> ()>
638
637
- !LinalgOperandDefConfig
639
638
name : rhs
640
639
kind : input_tensor
641
- type_var : T
640
+ type_var : T1
642
641
shape_map : affine_map<() -> ()>
643
642
- !LinalgOperandDefConfig
644
- name : out
643
+ name : O
645
644
kind : output_tensor
646
- type_var : T
645
+ type_var : T1
647
646
shape_map : affine_map<() -> ()>
648
647
indexing_maps : !LinalgIndexingMapsConfig
649
648
static_indexing_maps :
@@ -653,7 +652,7 @@ structured_op: !LinalgStructuredOpConfig
653
652
iterator_types : []
654
653
assignments :
655
654
- !ScalarAssign
656
- arg : out
655
+ arg : O
657
656
value : !ScalarExpression
658
657
scalar_fn :
659
658
kind : binary
@@ -668,14 +667,10 @@ metadata: !LinalgOpMetadata
668
667
name : matmul
669
668
cpp_class_name : MatmulOp
670
669
doc : |-
671
- Performs an accumulative (C += A*B) matrix multiplication of two 2D inputs.
670
+ Performs a matrix multiplication of two 2D inputs.
672
671
673
672
Numeric casting is performed on the operands to the inner multiply, promoting
674
673
them to the same data type as the accumulator/output.
675
-
676
- Note this operation represents an accumulative matrix multiplication, `C +=
677
- A*B` (as opposed to `C = A*B`). The appropriate initialisation of the output
678
- matrix should be done prior to calling this Op.
679
674
implements :
680
675
- LinalgContractionOpInterface
681
676
structured_op : !LinalgStructuredOpConfig
@@ -1141,8 +1136,8 @@ metadata: !LinalgOpMetadata
1141
1136
name : batch_mmt4d
1142
1137
cpp_class_name : BatchMmt4DOp
1143
1138
doc : |-
1144
- Performs a batched matrix-matrix-transpose multiplication of two batched-4D
1145
- (5D) inputs.
1139
+ Performs a batched matrix-matrix-transpose multiplication of two
1140
+ batched-4D (5D) inputs.
1146
1141
1147
1142
Besides the outermost batch dimension has the same semantic as
1148
1143
linalg.batch_matmul, the differences from linalg.batch_matmul in the
0 commit comments