ecsimsw
TF_source ) KernelRegistry 본문
///"tensorflow/core/framework/op_kernel.cc"
typedef std::unordered_multimap<string, KernelRegistration> KernelRegistry;
void* GlobalKernelRegistry() {
static KernelRegistry* global_kernel_registry = new KernelRegistry;
return global_kernel_registry;
}
static KernelRegistry* GlobalKernelRegistryTyped() {
return reinterpret_cast<KernelRegistry*>(GlobalKernelRegistry());
}
/// Conclusion
KernelRegistry : Unordered multi_map , includes key, kernelRegistration
-> 어디서 어떻게 사용되는거지, key랑 kernelRegistration를 insert했다가 언제 사용하는거지
'Machine Learning > tf_source' 카테고리의 다른 글
| TF_source ) kernel_builder (0) | 2019.04.23 |
|---|---|
| What's the difference between user registers and kernel registers? (0) | 2019.04.20 |
| TF_source) Opkernel (0) | 2019.04.12 |
| TF_source) OpkernelConstruction (0) | 2019.04.12 |
| TF_source ) Factory (0) | 2019.04.08 |
Comments