Searched refs:bn_op (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_batch_normalization.cc | 36 const auto* bn_op = in Run() local
39 auto& mean_array = model->GetArray(bn_op->inputs[1]); in Run()
40 const auto& multiplier_array = model->GetArray(bn_op->inputs[2]); in Run()
41 const auto& offset_array = model->GetArray(bn_op->inputs[3]); in Run()
43 CHECK(IsConstantParameterArray(*model, bn_op->inputs[1]) && in Run()
44 IsConstantParameterArray(*model, bn_op->inputs[2]) && in Run()
45 IsConstantParameterArray(*model, bn_op->inputs[3])) in Run()
66 AvailableArrayName(*model, bn_op->outputs[0] + "_mul"); in Run()
68 AvailableArrayName(*model, bn_op->outputs[0] + "_add"); in Run()
71 mul_op->inputs = {bn_op->inputs[0], mul_param_name}; in Run()
[all …]
|
| /external/tensorflow/tensorflow/contrib/quantize/python/ |
| D | fold_batch_norms.py | 86 match.variance_tensor + match.bn_op.get_attr('epsilon'))
234 bn_op = match_result.get_op(batch_norm_pattern)
236 bn_id_op = bn_op
238 batch_epsilon = bn_op.get_attr('epsilon')
242 output_tensor = bn_op.outputs[0]
266 is_training = bn_op.get_attr('is_training')
273 bn_op._set_attr(
277 mean_tensor = bn_op.outputs[1]
283 scope, sep, _ = bn_op.name.rpartition('/')
290 bn_op.outputs[2], (n - 1) / n, name='Undo_Bessel_Correction')
[all …]
|