1 /* Copyright 2020 The TensorFlow Authors. All Rights Reserved. 2 3 Licensed under the Apache License, Version 2.0 (the "License"); 4 you may not use this file except in compliance with the License. 5 You may obtain a copy of the License at 6 7 http://www.apache.org/licenses/LICENSE-2.0 8 9 Unless required by applicable law or agreed to in writing, software 10 distributed under the License is distributed on an "AS IS" BASIS, 11 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 12 See the License for the specific language governing permissions and 13 limitations under the License. 14 ==============================================================================*/ 15 #ifndef TENSORFLOW_CORE_TPU_KERNELS_TPU_OP_CONSTS_H_ 16 #define TENSORFLOW_CORE_TPU_KERNELS_TPU_OP_CONSTS_H_ 17 18 #include "absl/base/attributes.h" 19 20 namespace tensorflow { 21 namespace tpu { 22 23 // Resource names in the ResourceMgr. 24 // 25 // Name of cache for compiled TPU ISA protos. CompilationCache is created by 26 // ConfigureDistributedTpuOp, so only the master has a CompilationCache. 27 ABSL_CONST_INIT extern const char kCompilationCacheResourceName[]; 28 // Name of base class allowing Execute Ops to look up ISA protos. 29 // CompiledProtoCache is created by InitializeHostForDistributedTpuOp, so each 30 // tpu_worker has a CompiledProtoCache. 31 ABSL_CONST_INIT extern const char kCompiledProtoCacheResourceName[]; 32 // Name of cache unloader for compiled TPU ISA protos. Cache unloader should be 33 // put into TPU_SYSTEM device resource manager. Inference may use it to unload 34 // cache entries created during lifetime of a DirectSession. 35 ABSL_CONST_INIT extern const char kCompilationCacheUnloaderResourceName[]; 36 // TBD 37 ABSL_CONST_INIT extern const char kFingerprintLookupResourceName[]; 38 39 } // namespace tpu 40 } // namespace tensorflow 41 #endif // TENSORFLOW_CORE_TPU_KERNELS_TPU_OP_CONSTS_H_ 42