| /third_party/ffmpeg/libavfilter/dnn/ |
| D | dnn_backend_tf.c | 322 TF_OperationDescription *op_desc; in add_conv_layer() local
335 op_desc = TF_NewOperation(tf_model->graph, "Const", name_buffer); in add_conv_layer()
336 TF_SetAttrType(op_desc, "dtype", TF_FLOAT); in add_conv_layer()
344 TF_SetAttrTensor(op_desc, "value", tensor, tf_model->status); in add_conv_layer()
349 op = TF_FinishOperation(op_desc, tf_model->status); in add_conv_layer()
356 op_desc = TF_NewOperation(tf_model->graph, "Transpose", name_buffer); in add_conv_layer()
358 TF_AddInput(op_desc, input); in add_conv_layer()
360 TF_AddInput(op_desc, input); in add_conv_layer()
361 TF_SetAttrType(op_desc, "T", TF_FLOAT); in add_conv_layer()
362 TF_SetAttrType(op_desc, "Tperm", TF_INT32); in add_conv_layer()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/executor/tiling/ |
| D | op_tiling_adapter.cc | 74 … OpTilingCalculateAdapter::ConvertInputShapeAndType(const CNodePtr &node, ge::OpDescPtr *op_desc) { in ConvertInputShapeAndType() argument
76 MS_EXCEPTION_IF_NULL(*op_desc); in ConvertInputShapeAndType()
106 (void)(*op_desc)->AddInputDesc(input_name, ge_tensor_desc); in ConvertInputShapeAndType()
110 …OpTilingCalculateAdapter::ConvertOutputShapeAndType(const CNodePtr &node, ge::OpDescPtr *op_desc) { in ConvertOutputShapeAndType() argument
112 MS_EXCEPTION_IF_NULL(*op_desc); in ConvertOutputShapeAndType()
135 (void)(*op_desc)->AddOutputDesc(output_name, ge_tensor_desc); in ConvertOutputShapeAndType()
139 void OpTilingCalculateAdapter::ConvertCompileInfo(const CNodePtr &node, ge::OpDescPtr *op_desc) { in ConvertCompileInfo() argument
141 MS_EXCEPTION_IF_NULL(*op_desc); in ConvertCompileInfo()
144 (void)ge::AttrUtils::SetStr(*(*op_desc), COMPILE_INFO_KEY, compile_info_key); in ConvertCompileInfo()
145 (void)ge::AttrUtils::SetStr(*(*op_desc), COMPILE_INFO_JSON, op_compile_info_); in ConvertCompileInfo()
[all …]
|
| D | op_tiling_adapter.h | 44 void ConvertInputShapeAndType(const CNodePtr &node, ge::OpDescPtr *op_desc);
45 void ConvertOutputShapeAndType(const CNodePtr &node, ge::OpDescPtr *op_desc);
46 void ConvertCompileInfo(const CNodePtr &node, ge::OpDescPtr *op_desc);
47 void ConvertAttrs(const CNodePtr &node, ge::OpDescPtr *op_desc);
49 …ePtr &node, const std::map<uint32_t, tensor::TensorPtr> &depend_tensor_map, ge::OpDescPtr *op_desc,
|
| /third_party/mindspore/mindspore/ccsrc/runtime/hccl_adapter/ |
| D | converter.cc | 171 ge::OpDescPtr op_desc = std::make_shared<ge::OpDesc>(kStubDataStructureName, ge_node_name); in GenerateStubGeNode() local
172 MS_EXCEPTION_IF_NULL(op_desc); in GenerateStubGeNode()
180 op_desc->AddInputDesc( in GenerateStubGeNode()
189 (void)op_desc->AddOutputDesc( in GenerateStubGeNode()
195 …bool ret = ge::AttrUtils::SetDataType(*op_desc, ge::HCOM_ATTR_DATA_TYPE, ConvertHcclDTypeToGeDType… in GenerateStubGeNode()
202 (void)ConvertAttr<int64_t>(cnode, op_desc, kAttrRankSize, ge::HCOM_ATTR_RANK_SIZE); in GenerateStubGeNode()
203 auto group = ConvertAttr<std::string>(cnode, op_desc, kAttrGroup, ge::HCOM_ATTR_GROUP); in GenerateStubGeNode()
204 (void)ConvertAttr<int64_t>(cnode, op_desc, kAttrSrcRank, ge::HCOM_ATTR_SRC_RANK); in GenerateStubGeNode()
205 (void)ConvertAttr<int64_t>(cnode, op_desc, kAttrDestRank, ge::HCOM_ATTR_DEST_RANK); in GenerateStubGeNode()
206 (void)ConvertAttr<int64_t>(cnode, op_desc, kAttrSrTag, ge::HCOM_ATTR_SR_TAG); in GenerateStubGeNode()
[all …]
|
| /third_party/mesa3d/src/broadcom/qpu/ |
| D | qpu_pack.c | 594 const struct opcode_desc *op_desc) in opcode_invalid_in_version() argument
596 return (op_desc->first_ver != 0 && devinfo->ver < op_desc->first_ver) || in opcode_invalid_in_version()
597 (op_desc->last_ver != 0 && devinfo->ver > op_desc->last_ver); in opcode_invalid_in_version()
607 const struct opcode_desc *op_desc = &opcodes[i]; in lookup_opcode_from_packed() local
609 if (opcode < op_desc->opcode_first || in lookup_opcode_from_packed()
610 opcode > op_desc->opcode_last) in lookup_opcode_from_packed()
613 if (opcode_invalid_in_version(devinfo, op_desc)) in lookup_opcode_from_packed()
616 if (!(op_desc->mux_b_mask & (1 << mux_b))) in lookup_opcode_from_packed()
619 if (!(op_desc->mux_a_mask & (1 << mux_a))) in lookup_opcode_from_packed()
622 return op_desc; in lookup_opcode_from_packed()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/ |
| D | graph_kernel_splitter.cc | 60 bool IsRecomputeOp(const nlohmann::json &op_desc, const std::set<std::string> &recompute_ops) { in IsRecomputeOp() argument
61 std::vector<nlohmann::json> output_descs = op_desc[kJsonKeyOutputDesc]; in IsRecomputeOp()
101 void SetStitchAttr(const nlohmann::json &op_desc, const StitchInfo &info, const CNodePtr &node) { in SetStitchAttr() argument
102 std::vector<nlohmann::json> output_descs = op_desc[kJsonKeyOutputDesc]; in SetStitchAttr()
149 for (const auto &op_desc : op_node_descs) { in DecodeSplitNodes() local
150 … if (op_desc.find(kJsonKeyPtrAddress) == op_desc.end() || op_desc[kJsonKeyPtrAddress].is_null()) { in DecodeSplitNodes()
151 MS_LOG(ERROR) << "Decode failed, key: " << kJsonKeyPtrAddress << " not found in: " << op_desc; in DecodeSplitNodes()
155 std::string ptr_address = op_desc[kJsonKeyPtrAddress]; in DecodeSplitNodes()
161 if (IsRecomputeOp(op_desc, recompute_ops)) { in DecodeSplitNodes()
164 SetStitchAttr(op_desc, info, cp_node); in DecodeSplitNodes()
[all …]
|
| D | graph_kernel_helper.h | 81 …AnfToJsonDesc(const AnfNodePtrList &nodes, const DumpOption &dump_option, nlohmann::json *op_desc);
82 … AnfToJsonDesc(const AnfNodePtrList &nodes, const DumpOption &dump_option, nlohmann::json *op_desc,
84 …const std::vector<AnfNodePtrList> &graphs, const DumpOption &dump_option, nlohmann::json *op_desc);
|
| D | graph_kernel_helper.cc | 101 const DumpOption &dump_option, nlohmann::json *op_desc, in GenJson() argument
109 *op_desc = akg_kernel_json_generator.kernel_json(); in GenJson()
494 … AnfToJsonDesc(const AnfNodePtrList &nodes, const DumpOption &dump_option, nlohmann::json *op_desc, in AnfToJsonDesc() argument
496 MS_EXCEPTION_IF_NULL(op_desc); in AnfToJsonDesc()
519 return GenJson(op_nodes, in_and_out, dump_option, op_desc, address_node_map); in AnfToJsonDesc()
522 …nfToJsonDesc(const AnfNodePtrList &nodes, const DumpOption &dump_option, nlohmann::json *op_desc) { in AnfToJsonDesc() argument
523 MS_EXCEPTION_IF_NULL(op_desc); in AnfToJsonDesc()
547 return GenJson(op_nodes, in_and_out, dump_option, op_desc); in AnfToJsonDesc()
550 …onst std::vector<AnfNodePtrList> &graphs, const DumpOption &dump_option, nlohmann::json *op_desc) { in AnfToJsonDesc() argument
551 MS_EXCEPTION_IF_NULL(op_desc); in AnfToJsonDesc()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/ |
| D | tbe_json_creator.cc | 539 auto op_desc = AnfAlgo::GetOutputDataDesc(anf_node); in GenOutputDataDescJson() local
541 if (!op_desc.empty() && op_desc.at(0).find(kJListArgs) != op_desc.at(0).end()) { in GenOutputDataDescJson()
542 (*compute_json)[kJOutputDataDesc] = GetJsonValue<nlohmann::json>(op_desc.at(0), kJListArgs); in GenOutputDataDescJson()
|
| /third_party/mindspore/mindspore/_extends/graph_kernel/model/ |
| D | model_builder.py | 238 op_desc = map(dump_op_desc, self.desc[key])
239 desc[key] = [d for d in op_desc if d is not None]
|
| D | model.py | 470 input_desc, output_desc, op_desc = [], [], []
492 op_desc.append({'attr': attrs, 'impl_path': '',
496 … 'input_desc': input_desc, 'op': self.name, 'op_desc': op_desc, 'output_desc': output_desc,
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ |
| D | akg_kernel_json_decoder.cc | 412 for (const auto &op_desc : op_node_descs) { in DecodeFusedNodes() local
413 auto op_node = DecodeCNode(op_desc, graph, kernel_json[kJsonKeyProcess]); in DecodeFusedNodes()
|