HomeSort by relevance Sort by last modified time
    Searched refs:layer_op (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/contrib/quantize/python/
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,
222 layer_op: Matching conv/fc op prior to batch norm
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
1023 def layer_op(self): member in class:_BatchNormMatch
    [all...]
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
602 def layer_op(self): member in class:_LayerMatch
    [all...]

Completed in 59 milliseconds