Lines Matching full:kernel

27 namespace mindspore::kernel {
28 /// \brief The Kernel class is used to define a MindSpore Kernel.
29 class MS_API Kernel {
31   Kernel() = default;
34   /// \param[in] inputs define the input tensors for kernel.
35   /// \param[in] outputs define the output tensors for kernel.
36   /// \param[in] primitive define the primitive of kernel generated by flatbuffers.
37   /// \param[in] ctx define the context for kernel.
38 …Kernel(const std::vector<mindspore::MSTensor> &inputs, const std::vector<mindspore::MSTensor> &out…  in Kernel()  function
44   virtual ~Kernel() = default;
45   /// \brief prepare for executing kernel.
49   /// \brief execute the kernel.
53   /// \brief resize the kernel input shape, memory need to refresh.
57   /// \brief set kernel's input tensors.
61   /// \brief set kernel's input tensor.
66   /// \brief set kernel's output tensors.
70   /// \brief set kernel's output tensor.
75   /// \brief obtain kernel's input tensors.
79   /// \brief obtain kernel's output tensors.
83   /// \brief obtain kernel's name.
85   /// \return kernel's name.
87   /// \brief set kernel's name.
89   /// \param[in] name define the kernel's name.
91   /// \brief obtain kernel's context.
93   /// \return kernel's context.
95   /// \brief obtain kernel's type.
97   /// \return kernel's type.
99   /// \brief obtain the primitive of kernel generated by flatbuffers.
101   /// \return the primitive of kernel generated by flatbuffers.
103   /// \brief get kernel's attribute
105   /// \param[in] key define the kernel's attribute key.
115   /// \brief set kernel's attribute
117   /// \param[in] key define the kernel's attribute key.
118   /// \param[in] value define the kernel's attribute value.
133 }  // namespace mindspore::kernel