Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/acl/
Dacl_kernel_build.cc56 auto output_formats = build_info->GetAllOutputFormats(); in AclOpBuild() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/utils/
Dkernel_build_utils.cc61 void GetOutputFormat(const CNodePtr &kernel_node, std::vector<std::string> *output_formats) { in GetOutputFormat()
171const std::vector<std::string> &output_formats, const std::vector<TypeId> &output_types, in SetKernelBuildInfo()
193 std::vector<std::string> output_formats; in UpdateDynamicKernelBuildInfo() local
219 std::vector<std::string> output_formats; in UpdateCustomKernelBuildInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/acl_ir/
Dacl_helper.cc118 std::vector<std::string> output_formats{output_with_indexs.size(), format}; in SetParameterFormat() local
228 … const std::vector<std::string> &input_formats, std::vector<std::string> *output_formats) { in GetOutputBuildInfo()
260 void SetOutputIdentityFlag(const AnfNodePtr &node, const std::vector<std::string> &output_formats) { in SetOutputIdentityFlag()
267 std::vector<std::string> *output_formats) { in RefreshRefFormat()
485 std::vector<std::string> *output_formats, in GetValidKernelBuildInfo()
Dop_api_util.cc93 std::vector<std::string> *output_formats, in GetValidKernelBuildInfo()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/
Dkernel_select_cpu.cc101 void GetOutputFormat(const CNodePtr &kernel_node, std::vector<std::string> *output_formats) { in GetOutputFormat()
291const std::vector<std::string> &output_formats, const std::vector<TypeId> &output_types, in SetKernelBuildInfo()
306 std::vector<std::string> output_formats; in SetKernelBuildInfoWithSelectedAttr() local
424 std::vector<std::string> output_formats; in UpdateDynamicKernelBuildInfo() local
527 std::vector<std::string> output_formats; in UpdateCustomKernelBuildInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dgraph_kernel_helper.cc80const std::vector<std::string> &output_formats, in BuildSelectKernelBuildInfo()
89const std::vector<std::string> &output_formats, in BuildSelectKernelBuildInfo()
316 std::vector<std::string> output_formats = {out_info.format}; in CreateCNode() local
Dcast_matmul_fusion.cc34 std::vector<std::string> output_formats = AnfAlgo::GetAllOutputFormats(matmul_node); in UpdateBuildInfo() local
Ddecrease_compute_precision.cc78 std::vector<std::string> output_formats = AnfAlgo::GetAllOutputFormats(cnode); in UpdateOutputInfo() local
Dinsert_pad.cc202 std::vector<std::string> output_formats = AnfAlgo::GetAllOutputFormats(matmul_node); in UpdateMatmulInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/graph_ir/
Dcustom_op_infer.cc46 std::vector<int32_t> output_formats; in InferOffline() local
274 std::vector<int32_t> output_formats; in InferOnline() local
Dstorage_format_convertor.cc281 std::vector<std::string> output_formats; in InitParameterKernelInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/adapter/
Dcallback_impl.cc213 std::vector<std::string> output_formats; in SetBasicNodeKernelInfo() local
398 std::vector<std::string> output_formats; in SetBasicNodeKernelInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/graph_kernel/converter/
Dcallback_impl.cc175 std::vector<std::string> output_formats; in SetBasicNodeKernelInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/adapter/dpico/src/
Dcustom_creator.cc375 auto output_formats = subgraph->outputs_format; in SetCustomMultiOutput() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/cxx_api/
Dconverter_para.h48 std::vector<schema::Format> output_formats; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/
Dframework_utils.cc444const std::vector<std::string> &output_formats, const std::vector<TypeId> &output_types, in SetKernelBuildInfo()
886 const auto &output_formats = kernel_build_info->GetAllOutputFormats(); in UnfoldKernelBuildInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/config_parser/
Dconfig_file_parser.h130 std::string output_formats; // optional, default: NHWC member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dcpu_kernel.cc80 auto output_formats = outputs_ptr[j]->formats(); in GetSupportFromOpLib() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/device/
Dkernel_select_ascend.cc441 std::vector<std::string> output_formats; in GenerateKernelBuildInfo() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/pre_activate/pass/
Dinsert_type_transform_op_test.cc259 const std::vector<std::string> &output_formats, const std::vector<TypeId> &output_types, in SetKernelBuildInfo()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/ascend/model/
Dmodel_process.cc245 std::vector<Format> output_formats; in GetOutputFormat() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/hardware/
Dge_graph_executor.cc456 std::vector<std::string> output_formats; in SetKernelInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Dkernel_graph_mgr.cc470 const auto &output_formats = AnfAlgo::GetAllOutputFormats(node); in SaveAnfKernelInfo() local