| /external/tensorflow/tensorflow/compiler/xla/service/ |
| hlo_element_type_converter_test.cc | 112 ::testing::Matcher<const ::xla::HloInstruction*> batch_norm = local
115 op::Tuple(op::Convert(op::GetTupleElement(batch_norm, 0)),
116 op::Convert(op::GetTupleElement(batch_norm, 1)),
117 op::Convert(op::GetTupleElement(batch_norm, 2))));
|
| /external/tensorflow/tensorflow/python/layers/ |
| normalization.py | 790 batch_norm = batch_normalization variable
[all...] |
| /external/tensorflow/tensorflow/contrib/quantize/python/ |
| fold_batch_norms_test.py | 38 batch_norm = layers.batch_norm variable
97 normalizer_fn=batch_norm,
166 normalizer_fn=batch_norm,
229 normalizer_fn=batch_norm,
296 normalizer_fn=batch_norm,
375 normalizer_fn=batch_norm,
|
| quantize_parameterized_test.py | 33 batch_norm = layers.batch_norm variable
352 normalizer_fn=batch_norm,
435 normalizer_fn=batch_norm,
523 normalizer_fn=batch_norm,
|
| /external/tensorflow/tensorflow/contrib/layers/python/layers/ |
| layers.py | 59 'avg_pool2d', 'avg_pool3d', 'batch_norm', 'bias_add', 'conv2d', 'conv3d',
430 def batch_norm(inputs, function
[all...] |
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| virtual_scheduler_test.cc | 243 auto batch_norm = ops::FusedBatchNorm( local
246 auto y = batch_norm.y;
247 auto batch_mean = batch_norm.batch_mean;
248 auto batch_var = batch_norm.batch_variance;
806 auto batch_norm = ops::FusedBatchNorm( local
809 auto y = batch_norm.y;
810 auto batch_mean = batch_norm.batch_mean;
811 auto batch_var = batch_norm.batch_variance;
[all...] |
