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

  /external/tensorflow/tensorflow/contrib/quantize/python/
graph_matcher_test.py 15 """Tests for graph_matcher."""
24 from tensorflow.contrib.quantize.python import graph_matcher
62 inputs_pattern = graph_matcher.OpTypePattern('*', name='inputs')
63 relu_pattern = graph_matcher.OpTypePattern(
67 graph_matcher.OpTypePattern(
70 graph_matcher.OpTypePattern(
75 matcher = graph_matcher.GraphMatcher(relu_pattern)
97 y1_pattern = graph_matcher.OpTypePattern('*')
98 minus_pattern = graph_matcher.OpTypePattern('Sub', inputs=[y1_pattern, '*'])
99 matcher = graph_matcher.GraphMatcher(minus_pattern
    [all...]
quantize.py 23 from tensorflow.contrib.quantize.python import graph_matcher
360 input_pattern = graph_matcher.OpTypePattern('*')
361 weight_var_pattern = graph_matcher.OpTypePattern('Variable|VariableV2')
362 weight_partition_identity_pattern = graph_matcher.OpTypePattern(
364 weight_partition_concat_pattern = graph_matcher.OpTypePattern(
366 weight_identity_pattern = graph_matcher.OpTypePattern(
369 graph_matcher.OneofPattern([
375 weight_resource_var_pattern = graph_matcher.OpTypePattern('ReadVariableOp')
376 folded_weight_pattern = graph_matcher.OpTypePattern('Mul')
380 layer_pattern = graph_matcher.OpTypePattern
    [all...]
fold_batch_norms.py 23 from tensorflow.contrib.quantize.python import graph_matcher
151 input_pattern = graph_matcher.OpTypePattern('*')
154 weight_pattern = graph_matcher.OpTypePattern('*')
155 gamma_pattern = graph_matcher.OpTypePattern('*')
156 beta_pattern = graph_matcher.OpTypePattern('*')
157 mean_pattern = graph_matcher.OpTypePattern('*')
158 variance_pattern = graph_matcher.OpTypePattern('*')
160 moving_average_pattern = graph_matcher.OpTypePattern('*')
161 bn_decay_pattern = graph_matcher.OpTypePattern('*')
162 layer_pattern = graph_matcher.OpTypePattern
    [all...]

Completed in 80 milliseconds