Searched refs:a1_transpose_dims (Results 1 – 3 of 3) sorted by relevance
1369 llvm::SmallVector<int64_t, 4> a1_transpose_dims; in matchAndRewrite() local1370 a1_transpose_dims.push_back(filter_shape[1]); in matchAndRewrite()1371 a1_transpose_dims.push_back(filter_shape[2]); in matchAndRewrite()1372 a1_transpose_dims.push_back(filter_shape[3]); in matchAndRewrite()1373 a1_transpose_dims.push_back(filter_shape[0]); in matchAndRewrite()1376 a2_reshape_dims.push_back(a1_transpose_dims[0]); in matchAndRewrite()1377 a2_reshape_dims.push_back(a1_transpose_dims[1]); in matchAndRewrite()1378 a2_reshape_dims.push_back(a1_transpose_dims[2] / depth_multiplier.getInt()); in matchAndRewrite()1385 RankedTensorType::get(ArrayRef<int64_t>(a1_transpose_dims), in matchAndRewrite()
943 llvm::SmallVector<int64_t, 4> a1_transpose_dims; in matchAndRewrite() local944 a1_transpose_dims.push_back(filter_shape[2]); in matchAndRewrite()945 a1_transpose_dims.push_back(filter_shape[0]); in matchAndRewrite()946 a1_transpose_dims.push_back(filter_shape[1]); in matchAndRewrite()947 a1_transpose_dims.push_back(filter_shape[3]); in matchAndRewrite()952 RankedTensorType::get(ArrayRef<int64_t>(a1_transpose_dims), in matchAndRewrite()
2793 SmallVector<int64_t, 4> a1_transpose_dims; in convertTFConv2DCommon() local2794 a1_transpose_dims.push_back(filter_shape[3]); in convertTFConv2DCommon()2795 a1_transpose_dims.push_back(filter_shape[0]); in convertTFConv2DCommon()2796 a1_transpose_dims.push_back(filter_shape[1]); in convertTFConv2DCommon()2797 a1_transpose_dims.push_back(filter_shape[2]); in convertTFConv2DCommon()2802 RankedTensorType::get(ArrayRef<int64_t>(a1_transpose_dims), in convertTFConv2DCommon()