Home
last modified time | relevance | path

Searched defs:output_formats (Results 1 – 7 of 7) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dkernel_select_cpu.cc72 void GetOutputFormat(const CNodePtr &kernel_node, std::vector<std::string> *output_formats) { in GetOutputFormat()
102 … std::vector<std::string> *output_formats, std::vector<TypeId> *output_types) { in GetOutputFormatsAndDtypes()
179 … const std::vector<std::string> &output_formats, const std::vector<TypeId> &output_types, in SetKernelBuildInfo()
234 std::vector<std::string> output_formats; in UpdateDynamicKernelBuildInfoAndAttrs() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/graph_kernel/
Dcast_matmul_fusion.cc33 std::vector<std::string> output_formats = AnfAlgo::GetAllOutputFormats(matmul_node); in UpdateBuildInfo() local
Doptimize_assign.cc179 std::vector<std::string> output_formats = {input_formats.back()}; in ReplaceAssignByInplaceAssignInGraphkernel() local
Ddecrease_compute_precision.cc73 std::vector<std::string> output_formats = AnfAlgo::GetAllOutputFormats(cnode); in UpdateOutputInfo() local
Dgraph_kernel_helper.cc289 … const std::vector<std::string> &output_formats, in BuildSelectKernelBuildInfo()
305 … const std::vector<std::string> &output_formats, in BuildSelectKernelBuildInfo()
734 std::vector<std::string> output_formats = {out_info.format}; in CreateCNode() local
Dinsert_pad.cc226 std::vector<std::string> output_formats = AnfAlgo::GetAllOutputFormats(matmul_node); in UpdateMatmulInfo() local
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dcpu_kernel_factory.cc80 auto output_formats = outputs_ptr[j]->formats(); in SetKernelAttrs() local