| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/ |
| D | layout_optimization_to_nhwc.mlir | 61 "tf.FusedBatchNormV3"(%5, %arg1, %arg1, %arg1, %arg1)
69 // CHECK: "tf.FusedBatchNormV3"
102 "tf.FusedBatchNormV3"(%9, %arg2, %arg2, %arg2, %arg2)
109 // CHECK: %[[BATCH_NORM1:[_a-z0-9]*]], {{.*}} = "tf.FusedBatchNormV3"
129 "tf.FusedBatchNormV3"(%11, %arg2, %arg2, %arg2, %arg2)
136 // CHECK: %[[BATCH_NORM2:[_a-z0-9]*]], {{.*}} = "tf.FusedBatchNormV3"
|
| D | gpu_fusion.mlir | 9 …%y:6 = "tf.FusedBatchNormV3"(%arg0, %arg1, %arg2, %arg3, %arg4) {T = "tfdtype$DT_FLOAT", data_form…
18 …%y:6 = "tf.FusedBatchNormV3"(%arg0, %arg1, %arg2, %arg3, %arg4) {T = "tfdtype$DT_FLOAT", data_form…
31 …%y:6 = "tf.FusedBatchNormV3"(%arg0, %arg1, %arg2, %arg3, %arg4) {T = "tfdtype$DT_FLOAT", data_form…
43 …%y:6 = "tf.FusedBatchNormV3"(%arg0, %arg1, %arg2, %arg3, %arg4) {T = "tfdtype$DT_FLOAT", data_form…
|
| D | layout_optimization_layout_assignment_gpu_cc_60.mlir | 71 // CHECK: "tf.FusedBatchNormV3"
75 = "tf.FusedBatchNormV3"(%arg0, %arg1, %arg1, %arg1, %arg1)
96 // CHECK: "tf.FusedBatchNormV3"
100 = "tf.FusedBatchNormV3"(%arg0, %arg1, %arg1, %arg1, %arg1)
|
| D | layout_optimization_layout_assignment_gpu_cc_70.mlir | 152 // CHECK: "tf.FusedBatchNormV3"
156 = "tf.FusedBatchNormV3"(%arg0, %arg1, %arg1, %arg1, %arg1)
177 // CHECK: "tf.FusedBatchNormV3"
181 = "tf.FusedBatchNormV3"(%arg0, %arg1, %arg1, %arg1, %arg1)
|
| D | layout_optimization_move_transposes_end.mlir | 107 // CHECK: "tf.FusedBatchNormV3"(%arg0, {{.*}} {data_format = "NCHW"
115 // Compute FusedBatchNormV3 in NHWC format
117 = "tf.FusedBatchNormV3"(%1, %arg1, %arg1, %arg1, %arg1)
|
| D | layout_optimization_layout_assignment_to_nhwc.mlir | 47 // CHECK: "tf.FusedBatchNormV3"
59 = "tf.FusedBatchNormV3"(%arg0, %arg1, %arg1, %arg1, %arg1)
|
| D | layout_optimization_layout_assignment_to_nchw.mlir | 160 // CHECK: "tf.FusedBatchNormV3"
172 = "tf.FusedBatchNormV3"(%arg0, %arg1, %arg1, %arg1, %arg1)
|
| D | canonicalize.mlir | 1357 // CHECK: "tf.FusedBatchNormV3"
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/ |
| D | prepare-tf-with-allowing-bf16-and-f16-type-legalization.mlir | 14 …%0, %1, %2 ,%3, %4, %5 = "tf.FusedBatchNormV3"(%arg0, %arg1, %arg2, %arg3, %arg4) {T = "tfdtype$DT…
16 // CHECK: "tf.FusedBatchNormV3"
|
| D | prepare-tf.mlir | 68 …%0:6 = "tf.FusedBatchNormV3"(%arg0, %arg1, %arg2, %arg3, %arg4) {T = "tfdtype$DT_FLOAT", U = "tfdt…
71 …%1:6 = "tf.FusedBatchNormV3"( %0#0, %cst, %arg2, %arg3, %arg4) {T = "tfdtype$DT_FLOAT", U = "tfdty…
73 …%2:6 = "tf.FusedBatchNormV3"( %1#0, %cst, %arg2, %arg3, %arg4) {T = "tfdtype$DT_FLOAT", U = "tfdty…
75 …%3:6 = "tf.FusedBatchNormV3"( %2#0, %arg1, %arg2, %arg3, %arg4) {T = "tfdtype$DT_FLOAT", U = "tfdt…
97 // CHECK: %[[BATCHNORM1_a:[^,]+]], {{.*}} = "tf.FusedBatchNormV3"(%[[ADD2]], %[[CONSTANT1]], %[[AR…
98 // CHECK: %[[BATCHNORM1_b:[^,]+]], {{.*}} = "tf.FusedBatchNormV3"(%[[BATCHNORM1_a]], %[[CONSTANT1]…
99 // CHECK: "tf.FusedBatchNormV3"(%[[BATCHNORM1_b]], %[[ARG1]], %[[ARG2]], %[[ARG3]], %[[ARG4]])
724 …%result:6 = "tf.FusedBatchNormV3"(%arg0, %arg1, %arg2, %arg3, %arg4) {data_format = "NHWC", device…
726 // CHECK: "tf.FusedBatchNormV3"
745 …%0, %1, %2, %3, %4, %5 = "tf.FusedBatchNormV3"(%arg0, %arg1, %arg2, %arg3, %arg4) {data_format = "…
[all …]
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | FusedBatchNormV3.pbtxt | 2 name: "FusedBatchNormV3"
96 name: "FusedBatchNormV3"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | FusedBatchNormV3.pbtxt | 2 name: "FusedBatchNormV3"
103 name: "FusedBatchNormV3"
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_FusedBatchNormV3.pbtxt | 2 graph_op_name: "FusedBatchNormV3"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/compile_mlir_util/ |
| D | shape-inference-after-legalization.mlir | 6 …%0:6 = "tf.FusedBatchNormV3"(%arg0, %arg1, %arg1, %arg1, %arg1) {data_format = "NHWC", device = ""…
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_ex_op_test.cc | 162 ops::FusedBatchNormV3 fwd = ops::FusedBatchNormV3( in RunFusedBatchNorm()
164 ops::FusedBatchNormV3::IsTraining(is_training) in RunFusedBatchNorm()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_FusedBatchNormV3.pbtxt | 2 graph_op_name: "FusedBatchNormV3"
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper_test.cc | 150 auto fbn = ops::FusedBatchNormV3( in TEST_F()
152 ops::FusedBatchNormV3::IsTraining(is_training) in TEST_F()
261 auto fbn = ops::FusedBatchNormV3( in TEST_F()
263 ops::FusedBatchNormV3::IsTraining(is_training) in TEST_F()
|
| D | mkl_remapper_test.cc | 336 ops::FusedBatchNormV3(s.WithOpName("fused_batch_norm"), input_cast, in TEST_F()
338 ops::FusedBatchNormV3::IsTraining(is_training) in TEST_F()
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | op_select_allowlist.md | 196 * `raw_ops.FusedBatchNormV3`
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf.mlir | 54 …%0:6 = "tf.FusedBatchNormV3"(%arg0, %arg1, %arg2, %arg3, %arg4) {T = "tfdtype$DT_FLOAT", data_form…
63 …%0:6 = "tf.FusedBatchNormV3"(%arg0, %arg1, %arg2, %arg3, %arg4) {T = "tfdtype$DT_FLOAT", data_form…
74 …%0:6 = "tf.FusedBatchNormV3"(%arg0, %arg1, %arg2, %arg3, %arg4) {T = "tfdtype$DT_FLOAT", data_form…
86 …%0:6 = "tf.FusedBatchNormV3"(%arg0, %arg1, %arg2, %arg3, %arg4) {T = "tfdtype$DT_FLOAT", data_form…
95 …%0:6 = "tf.FusedBatchNormV3"(%arg0, %arg1, %arg2, %arg3, %arg4) {T = "tfdtype$DT_FLOAT", data_form…
120 …%0:6 = "tf.FusedBatchNormV3"(%arg0, %arg1, %arg2, %arg3, %arg4) {T = "tfdtype$DT_FLOAT", data_form…
128 …%0:6 = "tf.FusedBatchNormV3"(%arg0, %arg1, %arg2, %arg3, %arg4) {T = "tfdtype$DT_FLOAT", data_form…
135 …%0:6 = "tf.FusedBatchNormV3"(%arg0, %arg1, %arg2, %arg3, %arg4) {T = "tfdtype$DT_FLOAT", data_form…
141 // CHECK: tf.FusedBatchNormV3
142 …%0:6 = "tf.FusedBatchNormV3"(%arg0, %arg1, %arg2, %arg3, %arg4) {T = "tfdtype$DT_FLOAT", data_form…
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/g3doc/ |
| D | legalization.md | 1816 ### tf.FusedBatchNormV3
1823 %output = tf.FusedBatchNormV3(%x, %scale, %offset, %mean, %variance) {epsilon, data_format, is_trai…
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_tf.cc | 104 DECL_CONVERT_OP(FusedBatchNormV3);
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 2154 CreateFusedBatchNormOp<ops::FusedBatchNormV3>}; in TEST_P()
|
| /external/tensorflow/ |
| D | RELEASE.md | 1590 * Added support for `FusedBatchNormV3` in converter.
1710 * Added support for `FusedBatchNormV3` in converter.
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 1736 name: "FusedBatchNormV3"
|