Searched refs:GeTensorDesc (Results 1 – 12 of 12) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/ |
| D | op_adapter_util.cc | 137 return GeTensor(GeTensorDesc(ge::Shape({0}))); in VectorToTensorUtil()
190 GeTensorDesc desc(GeShape(), ge::FORMAT_NCHW, ge::DT_INT32); in ConvertAnyUtil()
197 GeTensorDesc desc(GeShape(), ge::FORMAT_NCHW, ge::DT_INT64); in ConvertAnyUtil()
204 GeTensorDesc desc(GeShape(), ge::FORMAT_NCHW, ge::DT_FLOAT); in ConvertAnyUtil()
211 GeTensorDesc desc(GeShape(), ge::FORMAT_NCHW, ge::DT_BOOL); in ConvertAnyUtil()
221 GeTensorDesc desc(shape, ge::FORMAT_NCHW, ge::DT_STRING); in ConvertAnyUtil()
|
| D | types.h | 48 using GeTensorDesc = ge::TensorDesc; variable
|
| D | op_adapter.h | 81 …std::shared_ptr<GeTensorDesc> CreateOutputDesc(const abstract::ShapePtr &shape_ptr, const TypePtr …
85 std::shared_ptr<GeTensorDesc> CreateNodeDesc(const AnfNodePtr &node, const std::string &format);
241 …std::shared_ptr<GeTensorDesc> CreateOutputDesc(const abstract::ShapePtr &shape_ptr, const TypePtr … in CreateOutputDesc()
251 std::shared_ptr<GeTensorDesc> CreateNodeDesc(const AnfNodePtr &node, const std::string &format) { in CreateNodeDesc()
|
| D | util.h | 76 …static std::shared_ptr<GeTensorDesc> GetGeTensorDesc(const ShapeVector &shape, const MeDataType &m…
|
| D | util.cc | 108 std::shared_ptr<GeTensorDesc> TransformUtil::GetGeTensorDesc(const ShapeVector &me_shape, const MeD… in GetGeTensorDesc()
136 auto desc = std::make_shared<GeTensorDesc>(shape, ge_format, data_type); in GetGeTensorDesc()
|
| D | op_adapter_base.h | 89 using UpdateOutputDescFunc = std::function<void(OperatorPtr, GeTensorDesc)>;
|
| D | convert.h | 118 …st(const std::shared_ptr<HcomBroadcast> &broadcast, const std::vector<GeTensorDesc> &broadcast_des…
|
| D | op_adapter.cc | 318 std::shared_ptr<GeTensorDesc> OpAdapterImpl::CreateOutputDesc(const abstract::ShapePtr &shape_ptr, … in CreateOutputDesc()
380 std::shared_ptr<GeTensorDesc> OpAdapterImpl::CreateNodeDesc(const AnfNodePtr &node, const std::stri… in CreateNodeDesc()
|
| D | convert.cc | 139 GeTensorDesc desc(GeShape(), ge::FORMAT_NCHW, ge::DT_INT64); in InitLoopVar()
305 const std::vector<GeTensorDesc> &broadcast_desc, in SetupBroadcast()
495 std::vector<GeTensorDesc> broadcast_desc; in GenerateBroadcastGraph()
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/op_declare/ |
| D | op_declare_macro.h | 74 [](const OperatorPtr op, const GeTensorDesc desc) { \
139 [](const OperatorPtr op, const GeTensorDesc desc) { \
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/tiling/ |
| D | op_tiling_adapter.cc | 100 ge::GeTensorDesc ge_tensor_desc; in ConvertInputShapeAndType()
129 ge::GeTensorDesc ge_tensor_desc; in ConvertOutputShapeAndType()
168 ge::GeTensorDesc ge_tensor_desc; in NewConstantOp()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hccl_adapter/ |
| D | converter.cc | 181 ge::GeTensorDesc(ge::GeShape(ge_shape), ge::Format::FORMAT_NCHW, in GenerateStubGeNode()
190 ge::GeTensorDesc(ge::GeShape(ge_shape), ge::Format::FORMAT_NCHW, in GenerateStubGeNode()
|