| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_json/ |
| D | single_tbe_json_creator.h | 51 void GenDescJson(const AnfNodePtr &anf_node, size_t node_out_idx, size_t desc_output_idx,
63 void GenDescJson(const AnfNodePtr &anf_node, size_t node_out_idx, size_t desc_output_idx,
|
| D | fusion_tbe_json_creator.cc | 192 GenDescJson(kernel_idx.first, kernel_idx.second, kernel_idx.second, &input_desc); in GenInputsJson()
261 GenDescJson(anf_node, i, desc_output_index[i], &output_desc); in GenOutputsJson()
273 GenDescJson(anf_node, i, i, &output_desc); in GenOutputsJson()
|
| D | tbe_json_creator.h | 87 virtual void GenDescJson(const AnfNodePtr &anf_node, size_t node_out_idx, size_t desc_output_idx,
|
| D | single_tbe_json_creator.cc | 180 GenDescJson(anf_node, node_out_idx, node_out_idx, output_desc); in GenOutputDescJson()
325 void SelectTbeJsonCreator::GenDescJson(const AnfNodePtr &anf_node, size_t node_out_idx, size_t desc… in GenDescJson() function in mindspore::kernel::SelectTbeJsonCreator
379 void CheckTbeJsonCreator::GenDescJson(const AnfNodePtr &anf_node, size_t node_out_idx, size_t desc_… in GenDescJson() function in mindspore::kernel::CheckTbeJsonCreator
|
| D | tbe_json_creator.cc | 393 void TbeJsonCreator::GenDescJson(const AnfNodePtr &anf_node, size_t node_out_idx, size_t desc_outpu… in GenDescJson() function in mindspore::kernel::TbeJsonCreator
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_build.h | 74 static void GenDescJson(const std::shared_ptr<mindspore::AnfNode> &anf_node, size_t node_out_idx,
|
| D | tbe_kernel_build.cc | 1095 void TbeKernelBuild::GenDescJson(const std::shared_ptr<mindspore::AnfNode> &anf_node, size_t node_o… in GenDescJson() function in mindspore::kernel::TbeKernelBuild
1169 GenDescJson(anf_node, node_out_idx, desc_output_idx, output_desc); in GenFusionOutputDescJson()
1285 GenDescJson(real_node, real_idx, real_idx, &output_desc, fusion_data_type); in GenFusionDataInputJson()
1402 GenDescJson(real_node, real_idx, real_idx, &input_desc); in GenFusionComputeInputJson()
|