| /external/tensorflow/tensorflow/contrib/quantize/python/ |
| D | fold_batch_norms_test.py | 75 fused_batch_norm, freeze_batch_norm_delay, argument
117 fused=fused_batch_norm)
131 scale=has_scaling, fused=fused_batch_norm),
145 self._BatchNormMultiplierName(conv_name, has_scaling, fused_batch_norm)
159 self._BathNormBiasName(conv_name, fused_batch_norm)
176 fused_batch_norm=False, argument
211 layer1, scale=has_scaling, fused=fused_batch_norm, scope='layer1')
222 scale=has_scaling, fused=fused_batch_norm),
226 layer2, scale=has_scaling, fused=fused_batch_norm, scope='layer2')
237 self._BatchNormMultiplierName(scope, has_scaling, fused_batch_norm)
[all …]
|
| D | quantize_parameterized_test.py | 490 with_bypass, delay, fused_batch_norm, argument
523 normalizer_params=self._BatchNormParams(fused_batch_norm),
547 with_bypass, delay, fused_batch_norm, argument
577 normalizer_params=self._BatchNormParams(fused_batch_norm),
604 fused_batch_norm, use_resource, scope): argument
636 normalizer_params=self._BatchNormParams(fused_batch_norm),
662 fused_batch_norm, use_resource, scope): argument
695 normalizer_params=self._BatchNormParams(fused_batch_norm),
740 fused_batch_norm, use_resource): argument
751 fused=fused_batch_norm, force_updates=True)
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 65 explicit FusedBatchNorm(const NodeDef* fused_batch_norm) in FusedBatchNorm()
66 : fused_batch_norm(fused_batch_norm) {} in FusedBatchNorm()
68 const NodeDef* fused_batch_norm = nullptr; member
108 Conv2DWithBatchNorm(const NodeDef* conv2d, const NodeDef* fused_batch_norm, in Conv2DWithBatchNorm()
110 : conv2d(conv2d), fused_batch_norm(fused_batch_norm), epsilon(epsilon) {} in Conv2DWithBatchNorm()
113 const NodeDef* fused_batch_norm = nullptr; member
121 const NodeDef* fused_batch_norm, in Conv2DWithBatchNormAndRelu()
124 fused_batch_norm(fused_batch_norm), in Conv2DWithBatchNormAndRelu()
129 const NodeDef* fused_batch_norm = nullptr; member
383 matched->fused_batch_norm = batch_norm; in FindConv2DWithBatchNorm()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_fused_impl.h | 326 FusedBatchNormArgs fused_batch_norm;
348 &fused_batch_norm));
350 fused_batch_norm.scaling_factor.data(),
351 fused_batch_norm.offset_data,
352 fused_batch_norm.estimated_mean_data),
359 &fused_batch_norm));
361 fusion_args.epsilon, fused_batch_norm.scaling_factor.data(),
362 fused_batch_norm.offset_data,
363 fused_batch_norm.estimated_mean_data),
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_fused_batchnorm_test.py | 77 y, _, _ = nn_impl.fused_batch_norm(
124 y, mean, var = nn_impl.fused_batch_norm(
204 y, _, _ = nn_impl.fused_batch_norm(
220 y32, _, _ = nn_impl.fused_batch_norm(
267 y, _, _ = nn_impl.fused_batch_norm(
308 y32, _, _ = nn_impl.fused_batch_norm(
|
| D | nn_impl.py | 1191 def fused_batch_norm( function
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
| D | vgg_block_nchw_test.py | 45 x, _, _ = nn_impl.fused_batch_norm(
|
| D | vgg_block_test.py | 45 x, _, _ = nn_impl.fused_batch_norm(
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | fused_batchnorm_test.py | 98 y, mean, variance = nn.fused_batch_norm(
141 y, mean, var = nn.fused_batch_norm(
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | normalization.py | 444 return nn.fused_batch_norm(
452 return nn.fused_batch_norm(
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | layout_optimizer_test.py | 91 y, mean, _ = nn.fused_batch_norm(x, scale, offset)
297 bn0 = nn.fused_batch_norm(split[0], scale, offset)
298 bn1 = nn.fused_batch_norm(split[1], scale, offset)
|
| /external/tensorflow/tensorflow/lite/g3doc/guide/ |
| D | ops_compatibility.md | 123 * [tf.nn.fused_batch_norm](https://www.tensorflow.org/api_docs/python/tf/nn/fused_batch_norm)
|
| /external/tensorflow/tensorflow/lite/ |
| D | build_def.bzl | 256 "fused_batch_norm",
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | mkl_layout_pass.cc | 265 csinfo_.fused_batch_norm = "FusedBatchNorm"; in MklLayoutRewritePass()
394 rinfo_.push_back({csinfo_.fused_batch_norm, in MklLayoutRewritePass()
395 mkl_op_registry::GetMklOpName(csinfo_.fused_batch_norm), in MklLayoutRewritePass()
697 string fused_batch_norm; member
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| D | layers.py | 368 return nn.fused_batch_norm(
372 return nn.fused_batch_norm(
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.nn.pbtxt | 212 name: "fused_batch_norm"
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 2234 return nn.fused_batch_norm(
2313 y, _, _ = nn.fused_batch_norm(
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | control_flow_ops_test.py | 364 outputs = nn.fused_batch_norm(
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | generate_examples.py | 1565 [x_norm, _, _] = tf.nn.fused_batch_norm(
|