Searched refs:conv2d_output (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op_test.cc | 418 Tensor conv2d_output, conv2d_meta_output; in VerifyFusedBatchNormGradWithConv2D() local
420 conv2d_test.RunConv2D(input, filter, &conv2d_output, in VerifyFusedBatchNormGradWithConv2D()
446 AddInputFromArray<T>(conv2d_output.shape(), conv2d_output.flat<T>()); in VerifyFusedBatchNormGradWithConv2D()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tfl.cc | 1066 Value conv2d_output; in matchAndRewrite() local
1068 conv2d_output = in matchAndRewrite()
1072 conv2d_output = a1_conv2d_op.getResult(); in matchAndRewrite()
1079 rewriter, op, conv2d_output, fused_activation_fn); in matchAndRewrite()
1087 rewriter.replaceOp(op, {conv2d_output}); in matchAndRewrite()
1200 Value conv2d_output; in matchAndRewrite() local
1202 conv2d_output = in matchAndRewrite()
1206 conv2d_output = a1_conv2d_op.getResult(); in matchAndRewrite()
1209 rewriter.replaceOp(op, {conv2d_output}); in matchAndRewrite()
1327 Value conv2d_output; in matchAndRewrite() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | legalize_hlo.cc | 370 OpResult conv2d_output = conv2d_op->getResult(0); in matchAndRewrite() local
371 auto conv2d_output_type = conv2d_output.getType().cast<ShapedType>(); in matchAndRewrite()
382 conv_op.getLoc(), output_permutation_and_shape.shape, conv2d_output, in matchAndRewrite()
|