Searched refs:GetInputReshapeType (Results 1 – 5 of 5) sorted by relevance
61 std::string GetInputReshapeType(size_t input_index) const;137 …kernel_build_info_->input_reshape_type_.emplace_back(kernel_build_info->GetInputReshapeType(index)… in KernelBuildInfoBuilder()
68 std::string KernelBuildInfo::GetInputReshapeType(size_t input_index) const { in GetInputReshapeType() function in mindspore::kernel::KernelBuildInfo
202 std::string padding_axis = is_insert_input ? AnfAlgo::GetInputReshapeType(node, insert_index) in AddTransOpNodeToGraph()219 AnfAlgo::GetInputReshapeType(node, insert_index)); in AddTransOpNodeToGraph()
172 static std::string GetInputReshapeType(const AnfNodePtr &node, size_t output_idx);
897 infer_shape = trans::PaddingShape(infer_shape, format, GetInputReshapeType(node, input_idx)); in GetInputDeviceShape()902 std::string AnfRuntimeAlgorithm::GetInputReshapeType(const AnfNodePtr &node, size_t input_idx) { in GetInputReshapeType() function in mindspore::session::AnfRuntimeAlgorithm920 return build_info->GetInputReshapeType(input_idx); in GetInputReshapeType()