Searched defs:input_size_list (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/ |
| D | tbe_kernel_parallel_build.cc | 70 std::vector<size_t> input_size_list; in TbeOpParallelBuild() local
127 … const std::string &json_name, const std::vector<size_t> &input_size_list, in SaveTaskInfo()
219 const std::vector<size_t> &input_size_list, in SaveSameOpInfo()
232 const std::vector<size_t> &input_size_list, in SaveSameFusionOpInfo()
273 …ildManager::SearchInCache(const std::string &json_name, const std::vector<size_t> &input_size_list, in SearchInCache()
286 KernelModPtr ParallelBuildManager::GenKernelMod(const std::vector<size_t> &input_size_list, in GenKernelMod()
|
| D | tbe_kernel_parallel_build.h | 37 std::vector<size_t> input_size_list; member
|
| D | tbe_kernel_build.cc | 821 void GetInputSizeList(const nlohmann::json &input_json, std::vector<size_t> *input_size_list) { in GetInputSizeList()
879 … TbeKernelBuild::GetIOSize(const nlohmann::json &kernel_json, std::vector<size_t> *input_size_list, in GetIOSize()
892 void GetRealInputSize(const nlohmann::json &input_json, std::vector<size_t> *input_size_list, size_… in GetRealInputSize()
911 void GetInputSizeList2(const nlohmann::json &input_json, std::vector<size_t> *input_size_list) { in GetInputSizeList2()
974 …TbeKernelBuild::GetIOSize2(const nlohmann::json &kernel_json, std::vector<size_t> *input_size_list, in GetIOSize2()
1514 …elBuild::CalInputSize(const nlohmann::json &fusion_op_list, std::vector<size_t> *input_size_list) { in CalInputSize()
1568 … std::vector<size_t> *input_size_list, std::vector<size_t> *output_size_list) { in GetIOSize()
|
| D | ascend_kernel_compile.cc | 671 std::vector<size_t> input_size_list; in AscendFusionOpCompile() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/ |
| D | kernel_fusion.cc | 88 std::vector<size_t> input_size_list; in KernelFusion() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/ |
| D | aicpu_kernel_build.cc | 43 std::vector<size_t> *input_size_list) { in SetIOIputSize()
80 std::vector<size_t> input_size_list; in SetIOSize() local
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/akg/ |
| D | akg_kernel_json_generator.h | 105 const std::vector<size_t> &input_size_list() const { return input_size_list_; } in input_size_list() function
|
| /third_party/mindspore/mindspore/ccsrc/fl/server/ |
| D | parameter_aggregator.cc | 264 const std::vector<size_t> &input_size_list = server_kernel->GetInputSizeList(); in AssignMemory() local
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_kernel_runtime.cc | 380 const auto &input_size_list = kernel_mode->GetInputSizeList(); in FetchMemUnitSize() local
|