We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 80c8760 commit 610c6f0Copy full SHA for 610c6f0
paddle/fluid/pir/dialect/op_generator/op_build_gen.py
@@ -120,6 +120,16 @@
120
'UnsqueezeInferMeta',
121
'UnsqueezeWithXShapeInferMeta',
122
'ArrayPopInferMeta',
123
+ # backward.h
124
+ 'CrossEntropyGradInferMeta',
125
+ 'CrossEntropyGrad2InferMeta',
126
+ 'CrossEntropyWithSoftmaxGradInferMeta',
127
+ 'CSoftmaxWithCrossEntropyGradInferMeta',
128
+ 'LSTMGradInferMeta',
129
+ 'FFTC2RGradInferMeta',
130
+ 'GruGradInferMeta',
131
+ 'GruUnitGradInferMeta',
132
+ 'NllLossGradInferMeta',
133
}
134
135
_PREPARE_DATA_WITH_VECTOR_INT64_MTTABLE_ATTRIBUTE = {'FrobeniusNormOp'}
0 commit comments