Home
last modified time | relevance | path

Searched refs:PreOrder (Results 1 – 6 of 6) sorted by relevance

/external/llvm-project/mlir/tools/mlir-linalg-ods-gen/
Dmlir-linalg-ods-gen.cpp879 template <typename Lambda, bool PreOrder>
924 template <typename Lambda, bool PreOrder>
1049 template <typename Lambda, bool PreOrder>
1055 static_cast<const TensorExpr &>(expr).visit<Lambda, PreOrder>(callback); in visit()
1058 static_cast<const TensorUse &>(expr).visit<Lambda, PreOrder>(callback); in visit()
1073 template <typename Lambda, bool PreOrder>
1075 if (!PreOrder) in visit()
1078 ::visit<Lambda, PreOrder>(*e, callback); in visit()
1079 if (PreOrder) in visit()
1083 template <typename Lambda, bool PreOrder>
/external/tensorflow/tensorflow/core/grappler/utils/
Dtraversal.h44 static DfsCallbacks PreOrder(std::function<void(const NodeDef*)> pre) { in PreOrder() function
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dauto_mixed_precision.cc692 static DfsTypeCallbacks PreOrder(std::function<void(int)> pre) { in PreOrder() function
1476 DfsTypeCallbacks::PreOrder([&](int idx) { in FindFloat32TensorListOpClustersAndDenylistUnsafe()
1517 DfsTypeCallbacks::PreOrder([&](int idx) { in FindTensorListImplicitFloat32Edges()
1579 DfsTypeCallbacks::PreOrder([&](int idx) { in PropagateDenyFwdThroughClearAndInfer()
1596 DfsTypeCallbacks::PreOrder([&](int idx) { in PropagateDenyFwdThroughClearAndInfer()
1633 DfsTypeCallbacks::PreOrder( in AddClearAndInferToAllowIfBetweenAllow()
1651 DfsTypeCallbacks::PreOrder([&](int idx) { in AddClearAndInferToAllowIfBetweenAllow()
1691 DfsTypeCallbacks::PreOrder([&](int idx) { in PropagateAllowThroughClear()
Dcommon_subgraph_elimination.cc192 DfsCallbacks::PreOrder([&](const NodeDef* node) { in DedupComputations()
Dloop_optimizer.cc473 DfsCallbacks::PreOrder([&](const NodeDef* node) { in GetStackPushNodesToConvert()
Dmemory_optimizer.cc1340 DfsCallbacks::PreOrder([&](const NodeDef* node) { in FindAssignNodesToRelax()