Home
last modified time | relevance | path

Searched defs:input_dtypes (Results 1 – 12 of 12) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/ut/cpp/ops/
Dtest_ops_binary_op.h28 std::vector<TypePtr> input_dtypes = {kFloat16, kFloat16}; in GetBinaryOpDefaultCases() local
Dtest_ops_addc_op.cc37 std::vector<TypePtr> input_dtypes = {kFloat16, kFloat16, kFloat16, kFloat16}; in GetCases() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/
Dlaunch_mul.cc26 std::vector<TypeId> input_dtypes = {dtype_, dtype_}; in ObtainMulKernelGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/session/
Dsingle_kernel_graph.cc24 …const std::string &op_name, const std::vector<TypeId> &input_dtypes, const std::vector<ShapeVector… in ConstructKernelGraphBasedOnSingleOp()
/third_party/mindspore/mindspore-src/source/mindspore/lite/src/extendrt/cxx_api/llm_engine/
Dllm_engine_plugin.h42 std::vector<TypeId> input_dtypes; member
Dllm_engine_plugin.cc135 std::string input_dtypes; in InitInputOptions() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/common/debug/profiler/
Dprofiling_framework_data.h60 std::vector<std::string> input_dtypes; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/device/
Dlaunch_transdata.cc83 std::vector<TypeId> input_dtypes = {dtype_}; in ConstructKernelGraph() local
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/cxx_api/
Dconverter_para.h41 std::vector<TypeId> input_dtypes; member
/third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/config_parser/
Dconfig_file_parser.h123 std::string input_dtypes; member
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/
Dcpu_kernel.cc74 auto input_dtypes = inputs_ptr[j]->dtypes(); in GetSupportFromOpLib() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/
Dframework_utils.cc883 const auto &input_dtypes = kernel_build_info->GetAllInputDeviceTypes(); in UnfoldKernelBuildInfo() local