|
|
@ -33,7 +33,7 @@ void SetKernelInfo(const CNodePtr &apply_kernel_ptr);
|
|
|
|
class KernelAttr {
|
|
|
|
class KernelAttr {
|
|
|
|
public:
|
|
|
|
public:
|
|
|
|
using DataType = std::pair<TypeId, std::string>;
|
|
|
|
using DataType = std::pair<TypeId, std::string>;
|
|
|
|
KernelAttr() = default;
|
|
|
|
KernelAttr() : all_same_(0) {}
|
|
|
|
~KernelAttr() = default;
|
|
|
|
~KernelAttr() = default;
|
|
|
|
|
|
|
|
|
|
|
|
KernelAttr &AddInputAttr(const TypeId &ms_type, const std::string &format = kOpFormat_DEFAULT) {
|
|
|
|
KernelAttr &AddInputAttr(const TypeId &ms_type, const std::string &format = kOpFormat_DEFAULT) {
|
|
|
|