Searched defs:CompiledSubgraph (Results 1 – 2 of 2) sorted by relevance
105 struct CompiledSubgraph : public core::RefCounted { struct106 TpuCompilationCacheInterface* parent = nullptr; // Not owned.108 bool initialized = false;113 Status initialization_status;116 int64_t last_use = -1;119 std::string subgraph_key;144 std::unique_ptr<CompiledSubgraph> sharding_entry; argument145 std::unique_ptr<CompiledSubgraph> unsharding_entry; argument149 CompiledSubgraph* main_entry = nullptr; argument152 std::unique_ptr<TpuProgramGroupInterface> tpu_program_group;[all …]
136 struct CompiledSubgraph : public core::RefCounted { struct160 EntryRefImpl(XRTCompilationCache* parent, CompiledSubgraph* entry); argument