Home
last modified time | relevance | path

Searched defs:input_formats (Results 1 – 20 of 20) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/utils/
Dkernel_build_utils.cc83 void GetInputFormat(const CNodePtr &kernel_node, std::vector<std::string> *input_formats) { in GetInputFormat()
170 void SetKernelBuildInfo(const std::vector<std::string> &input_formats, const std::vector<TypeId> &i… in SetKernelBuildInfo()
191 std::vector<std::string> input_formats; in UpdateDynamicKernelBuildInfo() local
212 std::vector<std::string> input_formats; in UpdateCustomKernelBuildInfo() local
372 std::vector<std::string> input_formats; in SetKernelInfoWithMsg() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/acl/
Dacl_kernel_build.cc54 auto input_formats = build_info->GetAllInputFormats(); in AclOpBuild() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/adapter/
Dcallback_impl.cc54 std::vector<std::string> *input_formats) { in GetTypeAndFormats()
135std::vector<std::string> *input_formats, bool is_basic_node) { in CollectInputTypesAndFormats()
203 std::vector<std::string> input_formats; in SetBasicNodeKernelInfo() local
263 std::vector<std::string> input_formats; in ResetKernelInfoInputs() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/transform/acl_ir/
Dacl_helper.cc158 const std::vector<std::string> &input_formats) { in NeedNDOutput()
183 const GeAdapterInfoPtr &ge_info, std::vector<std::string> *input_formats, in GetInputBuildInfo()
228const std::vector<std::string> &input_formats, std::vector<std::string> *output_formats) { in GetOutputBuildInfo()
266 …at(const std::unordered_map<size_t, size_t> &ref_map, const std::vector<std::string> &input_format… in RefreshRefFormat()
484 …AclHelper::GetValidKernelBuildInfo(const AnfNodePtr &node, std::vector<std::string> *input_formats, in GetValidKernelBuildInfo()
Dop_api_util.cc92 …OpApiUtil::GetValidKernelBuildInfo(const AnfNodePtr &node, std::vector<std::string> *input_formats, in GetValidKernelBuildInfo()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/hal/device/
Dkernel_select_cpu.cc117 void GetInputFormat(const CNodePtr &kernel_node, std::vector<std::string> *input_formats) { in GetInputFormat()
290 void SetKernelBuildInfo(const std::vector<std::string> &input_formats, const std::vector<TypeId> &i… in SetKernelBuildInfo()
312 std::vector<std::string> input_formats; in SetKernelBuildInfoWithSelectedAttr() local
422 std::vector<std::string> input_formats; in UpdateDynamicKernelBuildInfo() local
520 std::vector<std::string> input_formats; in UpdateCustomKernelBuildInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/graph_kernel/
Dcast_matmul_fusion.cc29 std::vector<std::string> input_formats = AnfAlgo::GetAllInputFormats(matmul_node); in UpdateBuildInfo() local
Dinsert_pad.cc148 std::vector<std::string> input_formats = AnfAlgo::GetAllInputFormats(matmul); in InsertPad() local
200 std::vector<std::string> input_formats = AnfAlgo::GetAllInputFormats(matmul_node); in UpdateMatmulInfo() local
Ddecrease_compute_precision.cc73 std::vector<std::string> input_formats = AnfAlgo::GetAllInputFormats(cnode); in UpdateOutputInfo() local
Dgraph_kernel_helper.cc306 std::vector<std::string> input_formats; in CreateCNode() local
/third_party/ffmpeg/libavfilter/
Dvf_hwupload.c47 AVFilterFormats *input_formats = NULL; in hwupload_query_formats() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/pass/
Dflatten_value_sequence_in_pyexecute.cc89 const auto &input_formats = AnfAlgo::GetAllInputFormats(node); in Process() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/device/
Dkernel_select_ascend.cc440 std::vector<std::string> input_formats; in GenerateKernelBuildInfo() local
545 std::vector<std::string> input_formats; in GenerateKernelPacketBuildInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/cxx_api/
Dconverter_para.h44 std::vector<schema::Format> input_formats; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/
Dframework_utils.cc443 void SetKernelBuildInfo(const std::vector<std::string> &input_formats, const std::vector<TypeId> &i… in SetKernelBuildInfo()
885 const auto &input_formats = kernel_build_info->GetAllInputFormats(); in UnfoldKernelBuildInfo() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/config_parser/
Dconfig_file_parser.h126 std::string input_formats; // optional, default: NHWC member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dcpu_kernel.cc75 auto input_formats = inputs_ptr[j]->formats(); in GetSupportFromOpLib() local
/third_party/mindspore/mindspore-src/source/tests/ut/cpp/pre_activate/pass/
Dinsert_type_transform_op_test.cc258 …const AnfNodePtr &node, const std::vector<std::string> &input_formats, const std::vector<TypeId> &… in SetKernelBuildInfo()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/kernel/ascend/model/
Dmodel_process.cc223 std::vector<Format> input_formats; in GetInputFormat() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Dkernel_graph_mgr.cc466 const auto &input_formats = AnfAlgo::GetAllInputFormats(node); in SaveAnfKernelInfo() local