|
26 | 26 | int8: ['Conv2D', 'Conv3D', 'DepthwiseConv2dNative', #'FusedBatchNorm', 'FusedBatchNormV2','FusedBatchNormV3',
|
27 | 27 | 'MatMul', 'BatchMatMulV2', 'ConcatV2', 'MaxPool', 'MaxPool3D', 'AvgPool']
|
28 | 28 | uint8: ['Conv2D', 'Conv3D', 'DepthwiseConv2dNative', 'MatMul', 'BatchMatMulV2', 'ConcatV2', 'MaxPool', 'MaxPool3D', 'AvgPool']
|
29 |
| - bf16: ['Conv2D', 'Conv2DBackpropFilter', 'Conv2DBackpropInput', 'Conv3D', 'Conv3DBackpropFilterV2', |
30 |
| - 'Conv3DBackpropInputV2', 'DepthwiseConv2dNative', 'DepthwiseConv2dNativeBackpropFilter', |
31 |
| - 'DepthwiseConv2dNativeBackpropInput', 'MatMul', 'BatchMatMul', 'BatchMatMulV2', # allow_list |
32 |
| - 'Add', 'AddN', 'AddV2', 'AvgPool', 'AvgPool3D', 'AvgPool3DGrad', 'AvgPoolGrad', 'BiasAdd', |
33 |
| - 'BiasAddGrad', 'BiasAddV1', 'FusedBatchNormV2', 'FusedBatchNormGradV2', 'FusedBatchNormV3', |
34 |
| - 'FusedBatchNormGradV3', 'LeakyRelu', 'LeakyReluGrad', 'Mul', 'Sub', 'Elu', 'EluGrad', 'FloorDiv', |
35 |
| - '_FusedBatchNormEx', 'Log', 'Log1p', 'LogSoftmax', 'Prod', 'RealDiv', 'Reciprocal', 'Selu', |
36 |
| - 'SeluGrad', 'Sigmoid', 'SigmoidGrad', 'Softmax', 'Softplus', 'SoftplusGrad', 'Softsign', |
37 |
| - 'SoftsignGrad', 'Sqrt', 'Tanh', 'TanhGrad', #infer_list |
38 |
| - 'Abs', 'ArgMax', 'ArgMin', 'BatchToSpace', 'BatchToSpaceND', 'BroadcastTo', 'Ceil', 'CheckNumerics', |
39 |
| - 'ClipByValue', 'Concat', 'ConcatV2', 'DepthToSpace', 'DynamicPartition', 'DynamicStitch', |
40 |
| - 'EnsureShape', 'Enter', #'Range', |
41 |
| - 'Equal', 'Exit', 'ExpandDims', 'Fill', 'Floor', |
42 |
| - 'Gather', 'GatherNd', 'GatherV2', 'Greater', 'GreaterEqual', |
43 |
| - 'Identity', 'IsFinite', 'IsInf', 'IsNan', 'Less', 'LessEqual', |
44 |
| - 'Max', 'Maximum', 'MaxPool', 'MaxPool3D', 'MaxPool3DGrad', 'MaxPoolGrad', 'MaxPoolGradGrad', |
45 |
| - 'MaxPoolGradGradV2', 'MaxPoolGradV2', 'MaxPoolV2', 'Merge', 'Min', 'Minimum', 'MirrorPad', 'MirrorPadGrad', |
46 |
| - 'Neg', 'NextIteration', 'NotEqual', 'OnesLike', 'Pack', 'Pad', 'PadV2', 'PreventGradient', 'Rank', |
47 |
| - 'Relu', 'Relu6', 'Relu6Grad', 'ReluGrad', 'Reshape', |
48 |
| - 'ResizeNearestNeighbor', 'ResizeNearestNeighborGrad', |
49 |
| - 'Reverse', 'ReverseSequence', 'ReverseV2', #'Round', |
50 |
| - 'Select', 'SelectV2', 'Shape', 'ShapeN', 'Sign', |
51 |
| - 'Slice', 'Snapshot', 'SpaceToBatch', 'SpaceToBatchND', 'SpaceToDepth', 'Split', 'SplitV', 'Squeeze', |
52 |
| - 'StopGradient', 'StridedSlice', 'StridedSliceGrad', 'Switch', |
53 |
| - 'Tile', 'TopK', 'TopKV2', 'Transpose', 'Where', 'Unpack', 'ZerosLike'] #clear_list |
| 29 | + bf16: ["Conv2D", "Conv2DBackpropFilter", "Conv2DBackpropInput", "Conv3D", "Conv3DBackpropFilterV2", "Conv3DBackpropInputV2", |
| 30 | + "DepthwiseConv2dNative", "DepthwiseConv2dNativeBackpropFilter", "DepthwiseConv2dNativeBackpropInput", "GRUBlockCell", |
| 31 | + "AUGRUBlockCell", "MklGRU", "MklAUGRU", "MatMul", "BatchMatMul", "BatchMatMulV2", # allow_list |
| 32 | + "Add", "AddN", "AddV2", "AvgPool", "AvgPool3D", "AvgPool3DGrad", "AvgPoolGrad", "BiasAdd", "BiasAddGrad", "BiasAddV1", |
| 33 | + "Erf", "FusedBatchNormV2", "FusedBatchNormGradV2", "FusedBatchNormV3", "FusedBatchNormGradV3", "LeakyRelu", "LeakyReluGrad", |
| 34 | + "Mul", "Sub", "Elu", "EluGrad", "FloorDiv", "_FusedBatchNormEx", "Log", "Log1p", "LogSoftmax", "Prod", "RealDiv", "Reciprocal", |
| 35 | + "Selu", "SeluGrad", "Sigmoid", "SigmoidGrad", "Softmax", "Softplus", "SoftplusGrad", "Softsign", "SoftsignGrad", "Sqrt", |
| 36 | + "Tanh", "TanhGrad", #infer_list |
| 37 | + "Abs", "ArgMax","ArgMin","BatchToSpace","BatchToSpaceND","BroadcastTo","Ceil","CheckNumerics","ClipByValue","Concat","ConcatV2", |
| 38 | + "DepthToSpace","DynamicPartition","DynamicStitch","EnsureShape","Enter","Equal","Exit","ExpandDims","Fill","Floor","Gather", |
| 39 | + "GatherNd","GatherV2","Greater","GreaterEqual","Identity","IsFinite","IsInf","IsNan","Less","LessEqual","Max","Maximum","MaxPool", |
| 40 | + "MaxPool3D","MaxPool3DGrad","MaxPoolGrad","MaxPoolGradGrad","MaxPoolGradGradV2","MaxPoolGradV2","MaxPoolV2","Merge","Min","Minimum", |
| 41 | + "MirrorPad","MirrorPadGrad","Neg","NextIteration","NotEqual","OnesLike","Pack","Pad","PadV2","PreventGradient","Rank","Relu","Relu6", |
| 42 | + "Relu6Grad","ReluGrad","Reshape","ResizeNearestNeighbor","ResizeNearestNeighborGrad","Reverse","ReverseSequence","ReverseV2","Round", |
| 43 | + "Select","SelectV2","Shape","ShapeN","Sign","Slice","Snapshot","SpaceToBatch","SpaceToBatchND","SpaceToDepth","Split","SplitV","Squeeze", |
| 44 | + "StopGradient","StridedSlice","StridedSliceGrad","Switch","Tile","TopK","TopKV2","Transpose","Where","Unpack","ZerosLike" #clear list |
| 45 | + ] |
54 | 46 | fp32: ['*'] # '*' means all op types
|
55 | 47 |
|
56 | 48 | capabilities: &common_capabilities
|
|
0 commit comments