Home
last modified time | relevance | path

Searched refs:fused_batch_norm (Results 1 – 24 of 24) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc108 explicit FusedBatchNorm(int fused_batch_norm) in FusedBatchNorm()
109 : fused_batch_norm(fused_batch_norm) {} in FusedBatchNorm()
111 int fused_batch_norm = kMissingIndex; member
118 int fused_batch_norm = kMissingIndex; member
201 ContractionWithBatchNorm(int contraction, int fused_batch_norm, in ContractionWithBatchNorm()
204 fused_batch_norm(fused_batch_norm), in ContractionWithBatchNorm()
208 int fused_batch_norm = kMissingIndex; member
215 ContractionWithBatchNormAndActivation(int contraction, int fused_batch_norm, in ContractionWithBatchNormAndActivation()
218 fused_batch_norm(fused_batch_norm), in ContractionWithBatchNormAndActivation()
223 int fused_batch_norm = kMissingIndex; member
[all …]
/external/tensorflow/tensorflow/python/ops/
Dnn_fused_batchnorm_test.py82 y, _, _ = nn_impl.fused_batch_norm(
165 y, mean, var = nn_impl.fused_batch_norm(
246 y, _, _ = nn_impl.fused_batch_norm(
263 y32, _, _ = nn_impl.fused_batch_norm(
312 y, _, _ = nn_impl.fused_batch_norm(
354 y32, _, _ = nn_impl.fused_batch_norm(
684 y, _, _ = nn_impl.fused_batch_norm(
708 nn_impl.fused_batch_norm(x, scale, offset)
716 nn_impl.fused_batch_norm(x, scale, offset)
726 nn_impl.fused_batch_norm(
[all …]
Dnn_fused_batchnorm_d9m_test.py84 op_output = nn_impl.fused_batch_norm(
99 op_output_b = nn_impl.fused_batch_norm(
129 op_output = nn_impl.fused_batch_norm(
Dnn_fused_batchnorm_deterministic_test.py86 op_output = nn_impl.fused_batch_norm(
101 op_output_b = nn_impl.fused_batch_norm(
131 op_output = nn_impl.fused_batch_norm(
Dnn_impl.py1599 def fused_batch_norm( function
/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_op_test.cc243 Node* fused_batch_norm; in FusedBatchNormInference() local
255 .Finalize(g, &fused_batch_norm)); in FusedBatchNormInference()
282 Node* fused_batch_norm; in FusedBatchNormGrad() local
296 .Finalize(g, &fused_batch_norm)); in FusedBatchNormGrad()
Dfused_batch_norm_ex_op_test.cc600 Node* fused_batch_norm; in FusedBatchNormEx() local
615 .Finalize(g, &fused_batch_norm)); in FusedBatchNormEx()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_tf.cc897 ::mlir::Operation *fused_batch_norm, in matchAndRewrite()
900 Operation::operand_range mean(fused_batch_norm->getOperands()); in matchAndRewrite()
903 Operation::operand_range offset(fused_batch_norm->getOperands()); in matchAndRewrite()
904 Operation::operand_range x(fused_batch_norm->getOperands()); in matchAndRewrite()
905 Operation::operand_range scale(fused_batch_norm->getOperands()); in matchAndRewrite()
906 Operation::operand_range variance(fused_batch_norm->getOperands()); in matchAndRewrite()
912 dyn_cast_or_null<::mlir::TF::FusedBatchNormV3Op>(fused_batch_norm); in matchAndRewrite()
989 auto odsLoc = rewriter.getFusedLoc({fused_batch_norm->getLoc()}); in matchAndRewrite()
1178 rewriter.replaceOp(fused_batch_norm, replace_values); in matchAndRewrite()
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/
Dvgg_block_test.py36 x, _, _ = nn_impl.fused_batch_norm(
Dvgg_block_nchw_test.py36 x, _, _ = nn_impl.fused_batch_norm(
/external/tensorflow/tensorflow/lite/testing/op_tests/
Dfused_batch_norm.py57 [x_norm, _, _] = tf.compat.v1.nn.fused_batch_norm(
/external/tensorflow/tensorflow/compiler/tests/
Dfused_batchnorm_test.py112 y, mean, variance = nn.fused_batch_norm(
167 y, mean, var = nn.fused_batch_norm(
/external/tensorflow/tensorflow/lite/g3doc/guide/
Dops_compatibility.md75 * `tf.nn.fused_batch_norm`
/external/tensorflow/tensorflow/python/keras/layers/normalization/
Dlayer_normalization.py324 outputs, _, _ = nn.fused_batch_norm(
/external/tensorflow/tensorflow/lite/testing/
Dbuild_def.bzl65 "fused_batch_norm",
231 "fused_batch_norm",
Dgenerate_examples_lib.py85 from tensorflow.lite.testing.op_tests.fused_batch_norm import make_fused_batch_norm_tests
/external/tensorflow/tensorflow/python/grappler/
Dlayout_optimizer_test.py87 y, mean, _ = nn.fused_batch_norm(x, scale, offset)
314 bn0 = nn.fused_batch_norm(split[0], scale, offset)
315 bn1 = nn.fused_batch_norm(split[1], scale, offset)
1354 y, _, _ = nn.fused_batch_norm(conv3d, scale, offset, data_format='NDHWC')
Dauto_mixed_precision_test.py87 return nn_impl.fused_batch_norm(
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_layout_pass.cc271 csinfo_.fused_batch_norm = "FusedBatchNorm"; in MklLayoutRewritePass()
468 rinfo_.push_back({csinfo_.fused_batch_norm, in MklLayoutRewritePass()
469 mkl_op_registry::GetMklOpName(csinfo_.fused_batch_norm), in MklLayoutRewritePass()
879 string fused_batch_norm; member
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.nn.pbtxt224 name: "fused_batch_norm"
/external/tensorflow/tensorflow/python/eager/
Dforwardprop_test.py532 return nn_impl.fused_batch_norm(
/external/tensorflow/tensorflow/python/keras/
Dbackend.py2970 return nn.fused_batch_norm(
3052 y, _, _ = nn.fused_batch_norm(
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dcontrol_flow_ops_test.py735 outputs = nn.fused_batch_norm(
/external/tensorflow/
DRELEASE.md626 * (since v2.7) `tf.compat.v1.nn.fused_batch_norm` backprop to `offset`
1464 * `tf.compat.v1.nn.fused_batch_norm` backprop to `offset` when
5769 average computation in the `fused_batch_norm`. You should see
5770 significant performance improvements when using `fused_batch_norm` in