Searched refs:layer_op (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/contrib/quantize/python/ |
| D | fold_batch_norms.py | 76 scope, sep, _ = match.layer_op.name.rpartition('/')
103 if match.layer_op.type == 'DepthwiseConv2dNative':
123 match.layer_op, match.input_tensor, scaled_weight_tensor,
231 layer_op = match_result.get_op(layer_pattern)
243 if layer_op.type == 'MatMul':
306 return layer_op, _BatchNormMatch(
307 layer_op=layer_op,
328 layer_op, layer_match = _GetLayerMatch(match_result)
329 if layer_op is not None:
330 if layer_op not in matched_layer_set:
[all …]
|
| D | quantize.py | 90 context = _GetContextFromOp(layer_match.layer_op)
485 layer_op = match_result.get_op(layer_pattern)
498 if layer_op not in matched_layer_set:
499 matched_layer_set.add(layer_op)
501 _LayerMatch(layer_op, weight_tensor, activation_op, bypass_op,
508 layer_op = match_result.get_op(layer_pattern)
519 if layer_op not in matched_layer_set:
521 matched_layer_set.add(layer_op)
523 _LayerMatch(layer_op, weight_tensor, activation_op, bypass_op, None,
533 layer_op = match_result.get_op(layer_pattern)
[all …]
|