Searched refs:OpTypePattern (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | pattern_utils.h | 128 struct OpTypePattern { struct
132 std::vector<OpTypePattern> children; argument
136 for (const OpTypePattern& child : children) { in DebugString() argument
185 bool GetMatchedNodes(const OpTypePattern& pattern,
198 bool DoesOpTypePatternMatch(const OpTypePattern& pattern,
|
| D | pattern_utils_test.cc | 66 OpTypePattern GetMatMulBiasAddGeluPattern() { in GetMatMulBiasAddGeluPattern()
70 OpTypePattern pattern_syntax{"Mul", "my_gelu", NodeStatus::kReplace, in GetMatMulBiasAddGeluPattern()
162 OpTypePattern pattern{"E", "my_e", NodeStatus::kReplace, in TEST_F()
236 OpTypePattern pattern{"E", "my_e", NodeStatus::kReplace, in TEST_F()
336 OpTypePattern pattern{"E", "my_e", NodeStatus::kReplace, in TEST_F()
371 OpTypePattern pattern = GetMatMulBiasAddGeluPattern(); in TEST_F()
407 OpTypePattern pattern = GetMatMulBiasAddGeluPattern(); in TEST_F()
428 OpTypePattern pattern = GetMatMulBiasAddGeluPattern(); in TEST_F()
529 OpTypePattern pattern{op, "my_e", NodeStatus::kReplace, in TEST_F()
|
| D | pattern_utils.cc | 58 const OpTypePattern& pattern, MutableNodeView* node_view, in DoesOpTypePatternMatch()
152 const OpTypePattern& child_pattern = in DoesOpTypePatternMatch()
169 const OpTypePattern& pattern, in GetMatchedNodes()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | transform_utils.h | 147 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,
|
| D | transform_utils.cc | 266 string OpTypePattern::DebugString() const { in DebugString()
268 for (const OpTypePattern& input : inputs) { in DebugString()
291 Status GraphMatcher::GetOpTypeMatches(const OpTypePattern& pattern, in GetOpTypeMatches()
309 const NodeDef& node, const OpTypePattern& pattern, in DoesOpTypeMatch()
352 const OpTypePattern& input_pattern = pattern.inputs[i]; in DoesOpTypeMatch()
364 const GraphDef& input_graph_def, const OpTypePattern& pattern, in ReplaceMatchingOpTypes()
|
| D | remove_nodes.cc | 56 OpTypePattern pattern = {op}; in RemoveNodes()
|
| D | sparsify_gather.cc | 226 const TransformFuncContext& context, const OpTypePattern& pattern, in SparsifyGatherInternal()
564 const OpTypePattern gather_pattern = in SparsifyGather()
575 const OpTypePattern gather_v2_pattern = in SparsifyGather()
|
| D | README.md | 854 this easy, the Graph Transform Tool provides the `OpTypePattern` syntax. This is
878 OpTypePattern conv_pattern({"Conv2D", {{"*"}, {"Const"}}});
885 OpTypePattern conv_pattern({
948 `ReplaceMatchingOpTypes` helper. This takes in a graph, an `OpTypePattern`
|
| D | quantize_nodes.cc | 539 OpTypePattern pattern = {"*"}; in HoistFakeQuants()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 1128 utils::OpTypePattern gelu_exact_pattern = in FindMatMulBiasAddAndGelu()
1167 utils::OpTypePattern subgraph_gpu = in FindMatMulBiasAddAndGelu()
1179 utils::OpTypePattern subgraph_cpu = in FindMatMulBiasAddAndGelu()
1200 utils::OpTypePattern* subgraph_pattern = in FindMatMulBiasAddAndGelu()
1204 utils::OpTypePattern gelu_approximate_pattern = in FindMatMulBiasAddAndGelu()
1323 utils::OpTypePattern mulmax_pattern{ in FindMulAndMaximum()
1412 utils::OpTypePattern sigmoidmul_pattern{ in FindSigmoidAndMul()
1480 utils::OpTypePattern layer_norm_pattern = in FindMklLayerNorm()
1978 utils::OpTypePattern fusion_pattern1 = in FindFusedBatchMatMul()
1991 utils::OpTypePattern fusion_pattern2 = in FindFusedBatchMatMul()
[all …]
|