Searched refs:fused_batch_norm_op (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | prepare_tf.cc | 1001 auto fused_batch_norm_op = in matchAndRewrite() local
1003 root = fused_batch_norm_op; in matchAndRewrite()
1004 x = fused_batch_norm_op.getODSOperands(0); in matchAndRewrite()
1005 scale = fused_batch_norm_op.getODSOperands(1); in matchAndRewrite()
1006 offset = fused_batch_norm_op.getODSOperands(2); in matchAndRewrite()
1007 mean = fused_batch_norm_op.getODSOperands(3); in matchAndRewrite()
1008 variance = fused_batch_norm_op.getODSOperands(4); in matchAndRewrite()
1013 if (!TFTypeIsFloat32Tensor(fused_batch_norm_op.x())) return failure(); in matchAndRewrite()
1017 fused_batch_norm_op->getAttrOfType<::mlir::FloatAttr>("epsilon"); in matchAndRewrite()
1024 fused_batch_norm_op, [&](::mlir::Diagnostic &diag) { in matchAndRewrite()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | BUILD | 319 "//tensorflow/core/kernels:fused_batch_norm_op",
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | BUILD | 1706 ":fused_batch_norm_op",
1839 ":fused_batch_norm_op",
1863 ":fused_batch_norm_op",
3887 ":fused_batch_norm_op",
3950 name = "fused_batch_norm_op",
3951 prefix = "fused_batch_norm_op",
5913 "fused_batch_norm_op.h",
6096 "fused_batch_norm_op.cc",
|