Home
last modified time | relevance | path

Searched refs:OpTypePattern (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/contrib/quantize/python/
Dgraph_matcher_test.py62 inputs_pattern = graph_matcher.OpTypePattern('*', name='inputs')
63 relu_pattern = graph_matcher.OpTypePattern(
67 graph_matcher.OpTypePattern(
70 graph_matcher.OpTypePattern(
97 y1_pattern = graph_matcher.OpTypePattern('*')
98 minus_pattern = graph_matcher.OpTypePattern('Sub', inputs=[y1_pattern, '*'])
121 add_or_sub_pattern = graph_matcher.OpTypePattern(
130 reshape_pattern = graph_matcher.OpTypePattern('Reshape')
132 graph_matcher.OpTypePattern(
136 graph_matcher.OpTypePattern(
[all …]
Dquantize.py360 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(
375 weight_resource_var_pattern = graph_matcher.OpTypePattern('ReadVariableOp')
376 folded_weight_pattern = graph_matcher.OpTypePattern('Mul')
380 layer_pattern = graph_matcher.OpTypePattern(
393 batch_to_space_pattern = graph_matcher.OpTypePattern(
397 graph_matcher.OpTypePattern('*'),
[all …]
Dfold_batch_norms.py151 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(
165 batch_to_space_pattern = graph_matcher.OpTypePattern(
[all …]
Dgraph_matcher.py37 class OpTypePattern(Pattern): class
70 if isinstance(input_pattern, Pattern) else OpTypePattern(input_pattern)
/external/tensorflow/tensorflow/tools/graph_transforms/
Dtransform_utils.h147 struct OpTypePattern { struct
149 std::vector<OpTypePattern> inputs; argument
170 Status GetOpTypeMatches(const OpTypePattern& pattern,
174 bool DoesOpTypeMatch(const NodeDef& node, const OpTypePattern& pattern,
199 const GraphDef& input_graph_def, const OpTypePattern& pattern,
Dtransform_utils.cc264 string OpTypePattern::DebugString() const { in DebugString()
266 for (const OpTypePattern& input : inputs) { in DebugString()
289 Status GraphMatcher::GetOpTypeMatches(const OpTypePattern& pattern, in GetOpTypeMatches()
307 const NodeDef& node, const OpTypePattern& pattern, in DoesOpTypeMatch()
350 const OpTypePattern& input_pattern = pattern.inputs[i]; in DoesOpTypeMatch()
362 const GraphDef& input_graph_def, const OpTypePattern& pattern, in ReplaceMatchingOpTypes()
Dsparsify_gather.cc227 const TransformFuncContext& context, const OpTypePattern& pattern, in SparsifyGatherInternal()
565 const OpTypePattern gather_pattern = in SparsifyGather()
576 const OpTypePattern gather_v2_pattern = in SparsifyGather()
Dremove_nodes.cc57 OpTypePattern pattern = {op}; in RemoveNodes()
DREADME.md856 this easy, the Graph Transform Tool provides the `OpTypePattern` syntax. This is
880 OpTypePattern conv_pattern({"Conv2D", {{"*"}, {"Const"}}});
887 OpTypePattern conv_pattern({
950 `ReplaceMatchingOpTypes` helper. This takes in a graph, an `OpTypePattern`
Dquantize_nodes.cc539 OpTypePattern pattern = {"*"}; in HoistFakeQuants()