Searched defs:KernelSpec (Results 1 – 3 of 3) sorted by relevance
53 struct TORCH_API KernelSpec { struct57 KernelSpec(const int64_t _key, const std::shared_ptr<Graph>& _graph) in KernelSpec() argument68 for (const auto n : graph_->nodes()) {82 int64_t key() const { in key()85 std::shared_ptr<Graph> graph() const { in graph()88 const Code& code() const { in code()91 int64_t nInputs() const { in nInputs()94 int64_t nTensorInputs() const { in nTensorInputs()98 std::vector<std::vector<int64_t>>& inputBroadcastGroups() { in inputBroadcastGroups()101 const std::vector<std::vector<int64_t>>& inputBroadcastGroups() const { in inputBroadcastGroups()[all …]
62 struct KernelSpec { struct76 Optional<KernelSpec> Find(const Signature& signature) const { in Find() argument
21114 "KernelSpec":{ object