Searched defs:GetSpecInputLayers (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ | ||
| D | tbe_adapter.cc | 293 bool TbeAdapter::GetSpecInputLayers(const std::string &op_name, const std::vector<mindspore::AnfNod… in GetSpecInputLayers() function in mindspore::kernel::tbe::TbeAdapter |
| D | tbe_kernel_build.cc | 1189 bool TbeKernelBuild::GetSpecInputLayers(const std::string &op_name, in GetSpecInputLayers() function in mindspore::kernel::TbeKernelBuild |