Searched refs:squeeze_axis (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | dilated_conv.h | 144 int64_t squeeze_axis = squeeze_dims[0].cast<IntegerAttr>().getInt(); in matchAndRewrite() local
145 if (squeeze_axis < 0) { in matchAndRewrite()
147 squeeze_axis += 4; in matchAndRewrite()
149 if (squeeze_axis != expand_axis) { in matchAndRewrite()
|
| /external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/ |
| D | test_manifest.txt | 1030 DISABLED_squeeze/squeeze_axis=None,dtype=tf.int32,input_shape=[1,2,1,3,1,4,1,1]
1031 DISABLED_squeeze/squeeze_axis=[],dtype=tf.int32,input_shape=[1,2,1,3,1,4,1,1]
1032 DISABLED_squeeze/squeeze_axis=[0,2],dtype=tf.int32,input_shape=[1,2,1,3,1,4,1,1]
1033 DISABLED_squeeze/squeeze_axis=[4,7],dtype=tf.int32,input_shape=[1,2,1,3,1,4,1,1]
1034 DISABLED_squeeze/squeeze_axis=[-1,0,2,0,7,-6],dtype=tf.int32,input_shape=[1,2,1,3,1,4,1,1]
1035 DISABLED_squeeze/squeeze_axis=[-1,-2,-4,-6,-8],dtype=tf.int32,input_shape=[1,2,1,3,1,4,1,1]
1036 DISABLED_squeeze/squeeze_axis=[0,2,4,6,7],dtype=tf.int32,input_shape=[1,2,1,3,1,4,1,1]
1037 DISABLED_squeeze/squeeze_axis=[7,6,4,2,0],dtype=tf.int32,input_shape=[1,2,1,3,1,4,1,1]
1038 DISABLED_squeeze/squeeze_axis=[6,6],dtype=tf.int32,input_shape=[1,2,1,3,1,4,1,1]
1039 squeeze/squeeze_axis=None,dtype=tf.float32,input_shape=[1,2,1,3,1,4,1,1]
[all …]
|