Searched refs:OpTypePattern (Results 1 – 9 of 9) 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, MutableNodeView* node_view,
196 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()
322 OpTypePattern pattern{"E", "my_e", NodeStatus::kReplace, in TEST_F()
357 OpTypePattern pattern = GetMatMulBiasAddGeluPattern(); in TEST_F()
393 OpTypePattern pattern = GetMatMulBiasAddGeluPattern(); in TEST_F()
414 OpTypePattern pattern = GetMatMulBiasAddGeluPattern(); in TEST_F()
|
| D | pattern_utils.cc | 28 const OpTypePattern& pattern, MutableNodeView* node_view, in DoesOpTypePatternMatch()
88 const OpTypePattern& child_pattern = pattern.children[i]; in DoesOpTypePatternMatch()
104 const OpTypePattern& pattern, MutableNodeView* node_view, 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 | 856 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`
|
| D | quantize_nodes.cc | 539 OpTypePattern pattern = {"*"}; in HoistFakeQuants()
|