Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/core/ir/
Dkernel_info_dev.h76 …void set_output_tensor_num(const ssize_t output_tensor_num) { output_tensor_num_ = output_tensor_n… in set_output_tensor_num()
77 ssize_t output_tensor_num() const { return output_tensor_num_; } in output_tensor_num() function
/third_party/mindspore/mindspore-src/source/mindspore/core/utils/
Danf_utils.cc307 ssize_t output_tensor_num = runtime_cache.runtime_cache().output_tensor_num(); in GetOutputTensorNum() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/graph_kernel/
Dgraph_kernel_json_generator.cc202 size_t output_tensor_num = AnfUtils::GetOutputTensorNum(cnode_); in ExtractOutputs() local
480 size_t output_tensor_num = AnfUtils::GetOutputTensorNum(anf_node); in CreateOutputDescJson() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/graph/
Dslice_prepose_pass.cc1521 auto output_tensor_num = GetOutputTensorNum(preceed_node); in Run() local