Searched refs:kPrimTile (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/irpass/ |
| D | tile_eliminate.h | 38 AnfVisitor::Match(prim::kPrimTile, {IsNode, IsVNode})(node); in operator()
|
| D | branch_culling.cc | 79 {prim::kPrimTile, {2}}, in InConvertWhiteList()
93 {prim::kPrimCumSum, {2}}, {prim::kPrimTile, {2}}, in InConvertWhiteList()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/optimizer/ |
| D | irpass.cc | 94 …iminate_ = MakeSubstitution(std::make_shared<TileEliminater>(), "tile_eliminate", prim::kPrimTile); in OptimizeIRPassLib()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_expander.cc | 60 {kAllTarget, OpLevel_0, prim::kPrimTile}, in GetExpandOps()
|
| D | reorder_ops.cc | 36 prim::kPrimSqueeze, prim::kPrimTile, prim::kPrimNeg, prim::kPrimRelu, in IsTypeInsensitive()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | primitive_infer_map.cc | 216 {prim::kPrimTile, {ops::TileInfer, nullptr, true}}, in GetPrimitiveToBackendEvalImplMap()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 211 inline const PrimitivePtr kPrimTile = std::make_shared<Primitive>(kTile); variable
|
| /third_party/mindspore/mindspore/ccsrc/transform/express_ir/ |
| D | onnx_exporter.cc | 1615 if (node->IsApply(prim::kPrimTile)) { in ExportCNode()
|