Searched defs:output_size_list (Results 1 – 9 of 9) sorted by relevance
71 std::vector<size_t> output_size_list; in TbeOpParallelBuild() local128 … const std::vector<size_t> &output_size_list, int64_t scope_id) { in SaveTaskInfo()220 const std::vector<size_t> &output_size_list) { in SaveSameOpInfo()233 const std::vector<size_t> &output_size_list) { in SaveSameFusionOpInfo()274 … const std::vector<size_t> &output_size_list, mindspore::AnfNode *node) const { in SearchInCache() argument287 const std::vector<size_t> &output_size_list, in GenKernelMod()
38 std::vector<size_t> output_size_list; member
849 void GetOutputSizeList(const nlohmann::json &output_json, std::vector<size_t> *output_size_list) { in GetOutputSizeList()880 std::vector<size_t> *output_size_list) { in GetIOSize()931 void GetRealOutputSize(const nlohmann::json &output_json, std::vector<size_t> *output_size_list, si… in GetRealOutputSize()950 void GetOutputSizeList2(const nlohmann::json &output_json, std::vector<size_t> *output_size_list) { in GetOutputSizeList2()975 std::vector<size_t> *output_size_list, const AnfNodePtr &anf_node) { in GetIOSize2()1532 std::vector<size_t> *output_size_list) { in CalOutputSize()1568 … std::vector<size_t> *input_size_list, std::vector<size_t> *output_size_list) { in GetIOSize()
672 std::vector<size_t> output_size_list; in AscendFusionOpCompile() local
89 std::vector<size_t> output_size_list; in KernelFusion() local
106 const std::vector<size_t> &output_size_list() const { return output_size_list_; } in output_size_list() function
348 const auto &output_size_list = kernel_mod->GetOutputSizeList(); in FetchOutputDeviceTensor() local
81 std::vector<size_t> output_size_list; in SetIOSize() local
381 const auto &output_size_list = kernel_mode->GetOutputSizeList(); in FetchMemUnitSize() local