Home
last modified time | relevance | path

Searched defs:mean_op (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/tools/graph_transforms/
Dfold_old_batch_norms_test.cc62 Output mean_op = in TestFoldOldBatchNorms() local
146 Output mean_op = in TestFoldOldBatchNormsAfterDepthwiseConv2dNative() local
224 Output mean_op = in TestFoldFusedBatchNorms() local
301 Output mean_op = in TestFoldFusedBatchNormsAfterDepthwiseConv2dNative() local
402 Output mean_op = in TestFoldFusedBatchNormsWithConcat() local
494 Output mean_op = in TestFoldFusedBatchNormsWithBatchToSpace() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_batch_norm_op_test.cc240 auto mean_op = in VerifyFusedBatchNorm() local
372 auto mean_op = in VerifyFusedBatchNormGradWithConv2D() local
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
Ddevice_transform_patterns.cc602 TFL::MeanOp mean_op, PatternRewriter& rewriter) const { in matchAndRewrite()
667 TFL::MeanOp mean_op, PatternRewriter& rewriter) const { in matchAndRewrite()