Home
last modified time | relevance | path

Searched defs:output_tensor_num (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/
Dkernel_query.cc38 size_t output_tensor_num = AnfAlgo::GetOutputTensorNum(kernel_node); in FilterInvalidKernelInfo() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/
Dakg_kernel_json_generator.cc80 size_t output_tensor_num = AnfAlgo::GetOutputTensorNum(cnode_); in ExtractOutputs() local
229 size_t output_tensor_num = AnfAlgo::GetOutputTensorNum(anf_node); in CreateOutputDescJson() local
/third_party/mindspore/mindspore/lite/tools/optimizer/graph/
Dslice_prepose_pass.cc1523 auto output_tensor_num = GetOutputTensorNum(preceed_node); in Run() local