Searched refs:kPrimShape (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | shape.h | 34 Shape() : PrimitiveC(prim::kPrimShape->name()) {} in Shape()
|
| D | shape.cc | 62 REGISTER_PRIMITIVE_EVAL_IMPL(Shape, prim::kPrimShape, ShapeInfer, ShapeInferValue, false);
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/ |
| D | tf_bidirection_gru_cf_fusion.cc | 86 auto fw_shape1 = VectorRef({GetPrim(prim::kPrimShape), input}); in DefineBidirectionRnnPattern()
109 auto fw_shape = VectorRef({GetPrim(prim::kPrimShape), input}); in DefineBidirectionRnnPattern()
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/fusion/ |
| D | constant_folding_fusion.cc | 63 CheckPrimitiveType(cnode, prim::kPrimShape)) { in GetCNodeInputTensors()
|
| D | tf_bidirection_gru_fusion.cc | 138 auto is_shape = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimShape)); in DefineFowardPattern()
208 auto is_shape = std::make_shared<CondVar>(std::bind(IsOpType, p1, prim::kPrimShape)); in DefinebackwardPattern()
|
| /third_party/mindspore/mindspore/lite/tools/converter/quantizer/ |
| D | quantize_util.cc | 126 …if (opt::CheckPrimitiveType(cnode, prim::kPrimShape) || opt::CheckPrimitiveType(cnode, prim::kPrim… in CanOpFullQuantized()
|
| /third_party/mindspore/mindspore/core/base/ |
| D | core_ops.h | 184 inline const PrimitivePtr kPrimShape = std::make_shared<Primitive>("Shape"); variable
|