Searched defs:scale_after_normalization (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | nn_batchnorm_test.py | 36 scale_after_normalization, shift_after_normalization): argument
42 scale_after_normalization, shift_after_normalization): argument
49 scale_after_normalization): argument
56 scale_after_normalization): argument
62 scale_after_normalization, shift_after_normalization): argument
116 scale_after_normalization, argument
|
| H A D | nn_impl.py | 1713 scale_after_normalization=None, argument
1768 scale_after_normalization, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | quantized_batch_norm_op.cc | 37 float variance_epsilon, bool scale_after_normalization, in ReferenceBatchNorm()
100 float variance_epsilon, bool scale_after_normalization, in FixedPointBatchNorm()
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tools/ |
| H A D | optimize_for_inference_lib.py | 198 def scale_after_normalization(node): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | fold_old_batch_norms.cc | 56 const bool scale_after_normalization = in GetScaleAndOffsetValues() local
|