fix C++ header guards.
PiperOrigin-RevId: 209679086
This commit is contained in:
parent
62fcb03449
commit
7817237d58
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_C_CHECKPOINT_READER_H
|
||||
#define TENSORFLOW_C_CHECKPOINT_READER_H
|
||||
#ifndef TENSORFLOW_C_CHECKPOINT_READER_H_
|
||||
#define TENSORFLOW_C_CHECKPOINT_READER_H_
|
||||
|
||||
#include <memory>
|
||||
#include <string>
|
||||
@ -79,4 +79,4 @@ class CheckpointReader {
|
||||
} // namespace checkpoint
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_C_CHECKPOINT_READER_H
|
||||
#endif // TENSORFLOW_C_CHECKPOINT_READER_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_C_TF_STATUS_HELPER_H
|
||||
#define TENSORFLOW_C_TF_STATUS_HELPER_H
|
||||
#ifndef TENSORFLOW_C_TF_STATUS_HELPER_H_
|
||||
#define TENSORFLOW_C_TF_STATUS_HELPER_H_
|
||||
|
||||
#include "tensorflow/c/c_api.h"
|
||||
#include "tensorflow/core/lib/core/status.h"
|
||||
@ -29,4 +29,4 @@ Status StatusFromTF_Status(const TF_Status* tf_status);
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_C_TF_STATUS_HELPER_H
|
||||
#endif // TENSORFLOW_C_TF_STATUS_HELPER_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMPILER_JIT_KERNELS_XLA_LOCAL_LAUNCH_OP_H_
|
||||
#define TENSORFLOW_COMPILER_JIT_KERNELS_XLA_LOCAL_LAUNCH_OP_H_
|
||||
#ifndef TENSORFLOW_COMPILER_JIT_KERNELS_XLA_LAUNCH_OP_H_
|
||||
#define TENSORFLOW_COMPILER_JIT_KERNELS_XLA_LAUNCH_OP_H_
|
||||
|
||||
#include "tensorflow/compiler/jit/xla_compilation_cache.h"
|
||||
#include "tensorflow/core/framework/allocator.h"
|
||||
@ -81,4 +81,4 @@ class XlaLocalLaunchOp : public XlaLocalLaunchBase {
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_COMPILER_JIT_KERNELS_XLA_LOCAL_LAUNCH_OP_H_
|
||||
#endif // TENSORFLOW_COMPILER_JIT_KERNELS_XLA_LAUNCH_OP_H_
|
||||
|
@ -167,4 +167,4 @@ xla::ScopedShapedBuffer ExtractSubShapedBuffer(
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif
|
||||
#endif // TENSORFLOW_COMPILER_JIT_XLA_LAUNCH_UTIL_H_
|
||||
|
@ -128,4 +128,4 @@ class XlaTensor {
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif
|
||||
#endif // TENSORFLOW_COMPILER_JIT_XLA_TENSOR_H_
|
||||
|
@ -12,8 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
#ifndef TENSORFLOW_COMPILER_TF2XLA_TPU_UTIL_H_
|
||||
#define TENSORFLOW_COMPILER_TF2XLA_TPU_UTIL_H_
|
||||
#ifndef TENSORFLOW_COMPILER_TF2XLA_SHARDING_UTIL_H_
|
||||
#define TENSORFLOW_COMPILER_TF2XLA_SHARDING_UTIL_H_
|
||||
|
||||
#include <string>
|
||||
|
||||
@ -48,4 +48,4 @@ void SetShardingDeviceAssignmentFromNode(const Node& src, Node* dst);
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_COMPILER_TF2XLA_TPU_UTIL_H_
|
||||
#endif // TENSORFLOW_COMPILER_TF2XLA_SHARDING_UTIL_H_
|
||||
|
@ -56,4 +56,4 @@ OpSharding Tuple(const ShapeTree<OpSharding>& shardings);
|
||||
} // namespace sharding_builder
|
||||
} // namespace xla
|
||||
|
||||
#endif
|
||||
#endif // TENSORFLOW_COMPILER_XLA_CLIENT_SHARDING_BUILDER_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_SERVICE_ITERATOR_UTIL_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_SERVICE_ITERATOR_UTIL_H_
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_ITERATOR_UTIL_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_ITERATOR_UTIL_H_
|
||||
|
||||
#include <iterator>
|
||||
#include <utility>
|
||||
@ -95,4 +95,4 @@ UnwrappingIterator<NestedIter> MakeUnwrappingIterator(NestedIter iter) {
|
||||
|
||||
} // namespace xla
|
||||
|
||||
#endif // TENSORFLOW_COMPILER_XLA_SERVICE_ITERATOR_UTIL_H_
|
||||
#endif // TENSORFLOW_COMPILER_XLA_ITERATOR_UTIL_H_
|
||||
|
@ -15,8 +15,8 @@ limitations under the License.
|
||||
|
||||
// Call graph for an HLO module.
|
||||
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_SERVICE_HLO_CALL_GRAPH_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_SERVICE_HLO_CALL_GRAPH_H_
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_SERVICE_CALL_GRAPH_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_SERVICE_CALL_GRAPH_H_
|
||||
|
||||
#include <ostream>
|
||||
|
||||
@ -272,4 +272,4 @@ class CallGraph {
|
||||
|
||||
} // namespace xla
|
||||
|
||||
#endif // TENSORFLOW_COMPILER_XLA_SERVICE_HLO_CALL_GRAPH_H_
|
||||
#endif // TENSORFLOW_COMPILER_XLA_SERVICE_CALL_GRAPH_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_SERVICE__CALL_INLINER_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_SERVICE__CALL_INLINER_H_
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_SERVICE_CALL_INLINER_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_SERVICE_CALL_INLINER_H_
|
||||
|
||||
#include <deque>
|
||||
|
||||
@ -42,4 +42,4 @@ class CallInliner : public HloPassInterface {
|
||||
|
||||
} // namespace xla
|
||||
|
||||
#endif // TENSORFLOW_COMPILER_XLA_SERVICE__CALL_INLINER_H_
|
||||
#endif // TENSORFLOW_COMPILER_XLA_SERVICE_CALL_INLINER_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_SERVICE_CPU_COPY_INSERTION_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_SERVICE_CPU_COPY_INSERTION_H_
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_SERVICE_CPU_CPU_COPY_INSERTION_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_SERVICE_CPU_CPU_COPY_INSERTION_H_
|
||||
|
||||
#include "tensorflow/compiler/xla/service/hlo_module.h"
|
||||
#include "tensorflow/compiler/xla/service/hlo_pass_interface.h"
|
||||
@ -39,4 +39,4 @@ class CpuCopyInsertion : public HloPassInterface {
|
||||
|
||||
} // namespace xla
|
||||
|
||||
#endif // TENSORFLOW_COMPILER_XLA_SERVICE_CPU_COPY_INSERTION_H_
|
||||
#endif // TENSORFLOW_COMPILER_XLA_SERVICE_CPU_CPU_COPY_INSERTION_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_SERVICE_CPU_TRANSFER_MANAGER_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_SERVICE_CPU_TRANSFER_MANAGER_H_
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_SERVICE_CPU_CPU_TRANSFER_MANAGER_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_SERVICE_CPU_CPU_TRANSFER_MANAGER_H_
|
||||
|
||||
#include <vector>
|
||||
|
||||
@ -76,4 +76,4 @@ class CpuTransferManager : public GenericTransferManager {
|
||||
|
||||
} // namespace xla
|
||||
|
||||
#endif // TENSORFLOW_COMPILER_XLA_SERVICE_CPU_TRANSFER_MANAGER_H_
|
||||
#endif // TENSORFLOW_COMPILER_XLA_SERVICE_CPU_CPU_TRANSFER_MANAGER_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_SERVICE_GPU_TRANSFER_MANAGER_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_SERVICE_GPU_TRANSFER_MANAGER_H_
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_SERVICE_GPU_GPU_TRANSFER_MANAGER_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_SERVICE_GPU_GPU_TRANSFER_MANAGER_H_
|
||||
|
||||
#include <vector>
|
||||
|
||||
@ -61,4 +61,4 @@ class GpuTransferManager : public GenericTransferManager {
|
||||
} // namespace gpu
|
||||
} // namespace xla
|
||||
|
||||
#endif // TENSORFLOW_COMPILER_XLA_SERVICE_GPU_TRANSFER_MANAGER_H_
|
||||
#endif // TENSORFLOW_COMPILER_XLA_SERVICE_GPU_GPU_TRANSFER_MANAGER_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_SERVICE_INTERPRETER_TRANSFER_MANAGER_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_SERVICE_INTERPRETER_TRANSFER_MANAGER_H_
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_SERVICE_INTERPRETER_INTERPRETER_TRANSFER_MANAGER_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_SERVICE_INTERPRETER_INTERPRETER_TRANSFER_MANAGER_H_
|
||||
|
||||
#include "tensorflow/compiler/xla/service/generic_transfer_manager.h"
|
||||
#include "tensorflow/core/platform/macros.h"
|
||||
@ -33,4 +33,4 @@ class InterpreterTransferManager : public GenericTransferManager {
|
||||
|
||||
} // namespace xla
|
||||
|
||||
#endif // TENSORFLOW_COMPILER_XLA_SERVICE_INTERPRETER_TRANSFER_MANAGER_H_
|
||||
#endif // TENSORFLOW_COMPILER_XLA_SERVICE_INTERPRETER_INTERPRETER_TRANSFER_MANAGER_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_SERVICE_CPU_KERNEL_SUPPORT_LIBRARY_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_SERVICE_CPU_KERNEL_SUPPORT_LIBRARY_H_
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_SERVICE_LLVM_IR_KERNEL_SUPPORT_LIBRARY_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_SERVICE_LLVM_IR_KERNEL_SUPPORT_LIBRARY_H_
|
||||
|
||||
#include <string>
|
||||
|
||||
@ -296,4 +296,4 @@ class KernelSupportLibrary {
|
||||
};
|
||||
} // namespace xla
|
||||
|
||||
#endif // TENSORFLOW_COMPILER_XLA_SERVICE_CPU_KERNEL_SUPPORT_LIBRARY_H_
|
||||
#endif // TENSORFLOW_COMPILER_XLA_SERVICE_LLVM_IR_KERNEL_SUPPORT_LIBRARY_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_SOURCE_MAP_UTIL_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_SOURCE_MAP_UTIL_H_
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_SERVICE_SOURCE_MAP_UTIL_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_SERVICE_SOURCE_MAP_UTIL_H_
|
||||
|
||||
#include "tensorflow/compiler/xla/service/executable.h"
|
||||
#include "tensorflow/compiler/xla/status.h"
|
||||
@ -43,4 +43,4 @@ Status InvalidParameterArgument(const OpMetadata& op_metadata,
|
||||
} // namespace source_map_util
|
||||
} // namespace xla
|
||||
|
||||
#endif // TENSORFLOW_COMPILER_XLA_SOURCE_MAP_UTIL_H_
|
||||
#endif // TENSORFLOW_COMPILER_XLA_SERVICE_SOURCE_MAP_UTIL_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMPLIER_XLA_TEST_H_
|
||||
#define TENSORFLOW_COMPLIER_XLA_TEST_H_
|
||||
#ifndef TENSORFLOW_COMPILER_XLA_TEST_H_
|
||||
#define TENSORFLOW_COMPILER_XLA_TEST_H_
|
||||
|
||||
// This header includes gmock.h and enables the use of gmock matchers in tests
|
||||
// in third_party/tensorflow/compiler/xla.
|
||||
@ -45,4 +45,4 @@ limitations under the License.
|
||||
|
||||
#include "tensorflow/core/platform/test.h"
|
||||
|
||||
#endif // TENSORFLOW_COMPLIER_XLA_TEST_H_
|
||||
#endif // TENSORFLOW_COMPILER_XLA_TEST_H_
|
||||
|
@ -12,8 +12,8 @@
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
// =============================================================================
|
||||
#ifndef TENSORFLOW_CONTRIB_LIB_UTILS_PARALLEL_FOR_H_
|
||||
#define TENSORFLOW_CONTRIB_LIB_UTILS_PARALLEL_FOR_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_BOOSTED_TREES_LIB_UTILS_PARALLEL_FOR_H_
|
||||
#define TENSORFLOW_CONTRIB_BOOSTED_TREES_LIB_UTILS_PARALLEL_FOR_H_
|
||||
|
||||
#include "tensorflow/core/lib/core/threadpool.h"
|
||||
|
||||
@ -30,4 +30,4 @@ void ParallelFor(int64 batch_size, int64 desired_parallelism,
|
||||
} // namespace boosted_trees
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_CONTRIB_LIB_UTILS_PARALLEL_FOR_H_
|
||||
#endif // TENSORFLOW_CONTRIB_BOOSTED_TREES_LIB_UTILS_PARALLEL_FOR_H_
|
||||
|
@ -12,8 +12,8 @@
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
// =============================================================================
|
||||
#ifndef TENSORFLOW_CONTRIB_LIB_UTILS_RANDOM_H_
|
||||
#define TENSORFLOW_CONTRIB_LIB_UTILS_RANDOM_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_BOOSTED_TREES_LIB_UTILS_RANDOM_H_
|
||||
#define TENSORFLOW_CONTRIB_BOOSTED_TREES_LIB_UTILS_RANDOM_H_
|
||||
|
||||
#include "tensorflow/core/lib/random/simple_philox.h"
|
||||
|
||||
@ -36,4 +36,4 @@ inline int32 PoissonBootstrap(random::SimplePhilox* rng) {
|
||||
} // namespace boosted_trees
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_CONTRIB_LIB_UTILS_RANDOM_H_
|
||||
#endif // TENSORFLOW_CONTRIB_BOOSTED_TREES_LIB_UTILS_RANDOM_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_CORE_KERNELS_CLOUD_BIGQUERY_PARTITION_ACCESSOR_H_
|
||||
#define TENSORFLOW_CORE_KERNELS_CLOUD_BIGQUERY_PARTITION_ACCESSOR_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_CLOUD_KERNELS_BIGQUERY_TABLE_ACCESSOR_H_
|
||||
#define TENSORFLOW_CONTRIB_CLOUD_KERNELS_BIGQUERY_TABLE_ACCESSOR_H_
|
||||
|
||||
#include <map>
|
||||
#include <memory>
|
||||
@ -198,4 +198,4 @@ class BigQueryTableAccessor {
|
||||
};
|
||||
|
||||
} // namespace tensorflow
|
||||
#endif // TENSORFLOW_CORE_KERNELS_CLOUD_BIGQUERY_PARTITION_ACCESSOR_H_
|
||||
#endif // TENSORFLOW_CONTRIB_CLOUD_KERNELS_BIGQUERY_TABLE_ACCESSOR_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_CORE_KERNELS_CLOUD_BIGQUERY_TABLE_ACCESSOR_TEST_DATA_H_
|
||||
#define TENSORFLOW_CORE_KERNELS_CLOUD_BIGQUERY_TABLE_ACCESSOR_TEST_DATA_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_CLOUD_KERNELS_BIGQUERY_TABLE_ACCESSOR_TEST_DATA_H_
|
||||
#define TENSORFLOW_CONTRIB_CLOUD_KERNELS_BIGQUERY_TABLE_ACCESSOR_TEST_DATA_H_
|
||||
|
||||
#include <string>
|
||||
|
||||
@ -401,4 +401,4 @@ const string kTestEmptyRow = R"({
|
||||
} // namespace
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_CORE_KERNELS_CLOUD_BIGQUERY_TABLE_ACCESSOR_TEST_DATA_H_
|
||||
#endif // TENSORFLOW_CONTRIB_CLOUD_KERNELS_BIGQUERY_TABLE_ACCESSOR_TEST_DATA_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef GDR_MEMORY_MANAGER_H_
|
||||
#define GDR_MEMORY_MANAGER_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_GDR_GDR_MEMORY_MANAGER_H_
|
||||
#define TENSORFLOW_CONTRIB_GDR_GDR_MEMORY_MANAGER_H_
|
||||
|
||||
#include "google/protobuf/any.pb.h"
|
||||
#include "tensorflow/core/lib/core/status.h"
|
||||
@ -57,4 +57,4 @@ RemoteMemoryManager* CreateRemoteMemoryManager(const string& host,
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // GDR_MEMORY_MANAGER_H_
|
||||
#endif // TENSORFLOW_CONTRIB_GDR_GDR_MEMORY_MANAGER_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef GDR_RENDEZVOUS_MGR_H_
|
||||
#define GDR_RENDEZVOUS_MGR_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_GDR_GDR_RENDEZVOUS_MGR_H_
|
||||
#define TENSORFLOW_CONTRIB_GDR_GDR_RENDEZVOUS_MGR_H_
|
||||
|
||||
#include "tensorflow/contrib/gdr/gdr_memory_manager.h"
|
||||
#include "tensorflow/core/distributed_runtime/base_rendezvous_mgr.h"
|
||||
@ -39,4 +39,4 @@ class GdrRendezvousMgr : public BaseRendezvousMgr {
|
||||
|
||||
} // end namespace tensorflow
|
||||
|
||||
#endif // GDR_RENDEZVOUS_MGR_H_
|
||||
#endif // TENSORFLOW_CONTRIB_GDR_GDR_RENDEZVOUS_MGR_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef GDR_SERVER_LIB_H_
|
||||
#define GDR_SERVER_LIB_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_GDR_GDR_SERVER_LIB_H_
|
||||
#define TENSORFLOW_CONTRIB_GDR_GDR_SERVER_LIB_H_
|
||||
|
||||
#include "tensorflow/contrib/gdr/gdr_memory_manager.h"
|
||||
#include "tensorflow/core/distributed_runtime/rpc/grpc_server_lib.h"
|
||||
@ -49,4 +49,4 @@ class GdrServer : public GrpcServer {
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // GDR_SERVER_LIB_H_
|
||||
#endif // TENSORFLOW_CONTRIB_GDR_GDR_SERVER_LIB_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef GDR_WORKER_H_
|
||||
#define GDR_WORKER_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_GDR_GDR_WORKER_H_
|
||||
#define TENSORFLOW_CONTRIB_GDR_GDR_WORKER_H_
|
||||
|
||||
#include "tensorflow/contrib/gdr/gdr_memory_manager.h"
|
||||
|
||||
@ -44,4 +44,4 @@ class GdrWorker : public GrpcWorker {
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // GDR_WORKER_H_
|
||||
#endif // TENSORFLOW_CONTRIB_GDR_GDR_WORKER_H_
|
||||
|
@ -39,4 +39,4 @@ template void resize<float>(float*, unsigned char*, int, int, int, int, int,
|
||||
} // namespace label_image
|
||||
} // namespace tflite
|
||||
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_EXAMPLES_LABEL_IMAGE_BITMAP_HELPERS_H
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_EXAMPLES_LABEL_IMAGE_BITMAP_HELPERS_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_EXAMPLES_LABEL_IMAGE_GET_TOP_N_H
|
||||
#define TENSORFLOW_CONTRIB_LITE_EXAMPLES_LABEL_IMAGE_GET_TOP_N_H
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_EXAMPLES_LABEL_IMAGE_GET_TOP_N_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_EXAMPLES_LABEL_IMAGE_GET_TOP_N_H_
|
||||
|
||||
#include "tensorflow/contrib/lite/examples/label_image/get_top_n_impl.h"
|
||||
|
||||
@ -35,4 +35,4 @@ template void get_top_n<float>(float*, int, size_t, float,
|
||||
} // namespace label_image
|
||||
} // namespace tflite
|
||||
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_EXAMPLES_LABEL_IMAGE_GET_TOP_N_H
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_EXAMPLES_LABEL_IMAGE_GET_TOP_N_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_EXAMPLES_LABEL_IMAGE_GET_TOP_N_IMPL_H
|
||||
#define TENSORFLOW_CONTRIB_LITE_EXAMPLES_LABEL_IMAGE_GET_TOP_N_IMPL_H
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_EXAMPLES_LABEL_IMAGE_GET_TOP_N_IMPL_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_EXAMPLES_LABEL_IMAGE_GET_TOP_N_IMPL_H_
|
||||
|
||||
#include <algorithm>
|
||||
#include <queue>
|
||||
@ -67,4 +67,4 @@ void get_top_n(T* prediction, int prediction_size, size_t num_results,
|
||||
} // namespace label_image
|
||||
} // namespace tflite
|
||||
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_EXAMPLES_LABEL_IMAGE_GET_TOP_N_IMPL_H
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_EXAMPLES_LABEL_IMAGE_GET_TOP_N_IMPL_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_EXAMPLES_LABEL_IMAGE_LABEL_IMAGE_H
|
||||
#define TENSORFLOW_CONTRIB_LITE_EXAMPLES_LABEL_IMAGE_LABEL_IMAGE_H
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_EXAMPLES_LABEL_IMAGE_LABEL_IMAGE_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_EXAMPLES_LABEL_IMAGE_LABEL_IMAGE_H_
|
||||
|
||||
#include "tensorflow/contrib/lite/string.h"
|
||||
|
||||
@ -40,4 +40,4 @@ struct Settings {
|
||||
} // namespace label_image
|
||||
} // namespace tflite
|
||||
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_EXAMPLES_LABEL_IMAGE_LABEL_IMAGE_H
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_EXAMPLES_LABEL_IMAGE_LABEL_IMAGE_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_JAVA_EXCEPTION_JNI_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_JAVA_EXCEPTION_JNI_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_JAVA_SRC_MAIN_NATIVE_EXCEPTION_JNI_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_JAVA_SRC_MAIN_NATIVE_EXCEPTION_JNI_H_
|
||||
|
||||
#include <jni.h>
|
||||
#include "tensorflow/contrib/lite/error_reporter.h"
|
||||
@ -47,4 +47,4 @@ class BufferErrorReporter : public tflite::ErrorReporter {
|
||||
#ifdef __cplusplus
|
||||
} // extern "C"
|
||||
#endif // __cplusplus
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_JAVA_EXCEPTION_JNI_H_
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_JAVA_SRC_MAIN_NATIVE_EXCEPTION_JNI_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_JAVA_NATIVEINTERPRETERWRAPPER_JNI_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_JAVA_NATIVEINTERPRETERWRAPPER_JNI_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_JAVA_SRC_MAIN_NATIVE_NATIVEINTERPRETERWRAPPER_JNI_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_JAVA_SRC_MAIN_NATIVE_NATIVEINTERPRETERWRAPPER_JNI_H_
|
||||
|
||||
#include <jni.h>
|
||||
#include <stdio.h>
|
||||
@ -230,4 +230,4 @@ JNIEXPORT void JNICALL Java_org_tensorflow_lite_NativeInterpreterWrapper_delete(
|
||||
#ifdef __cplusplus
|
||||
} // extern "C"
|
||||
#endif // __cplusplus
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_JAVA_NATIVEINTERPRETERWRAPPER_JNI_H_
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_JAVA_SRC_MAIN_NATIVE_NATIVEINTERPRETERWRAPPER_JNI_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_JAVA_TENSOR_JNI_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_JAVA_TENSOR_JNI_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_JAVA_SRC_MAIN_NATIVE_TENSOR_JNI_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_JAVA_SRC_MAIN_NATIVE_TENSOR_JNI_H_
|
||||
|
||||
#include <jni.h>
|
||||
#include "tensorflow/contrib/lite/context.h"
|
||||
@ -92,4 +92,4 @@ Java_org_tensorflow_lite_Tensor_writeMultiDimensionalArray(JNIEnv* env,
|
||||
#ifdef __cplusplus
|
||||
} // extern "C"
|
||||
#endif // __cplusplus
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_JAVA_TENSOR_JNI_H_
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_JAVA_SRC_MAIN_NATIVE_TENSOR_JNI_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_JAVA_TENSORFLOW_LITE_JNI_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_JAVA_TENSORFLOW_LITE_JNI_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_JAVA_SRC_MAIN_NATIVE_TENSORFLOW_LITE_JNI_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_JAVA_SRC_MAIN_NATIVE_TENSORFLOW_LITE_JNI_H_
|
||||
|
||||
#include <jni.h>
|
||||
|
||||
@ -33,4 +33,4 @@ Java_org_tensorflow_lite_TensorFlowLite_version(JNIEnv*, jclass);
|
||||
#ifdef __cplusplus
|
||||
} // extern "C"
|
||||
#endif // __cplusplus
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_JAVA_TENSORFLOW_LITE_JNI_H_
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_JAVA_SRC_MAIN_NATIVE_TENSORFLOW_LITE_JNI_H_
|
||||
|
@ -12,8 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_KERNELS_INTERNAL_OPTIMIZED_CPU_CHECK_
|
||||
#define TENSORFLOW_CONTRIB_LITE_KERNELS_INTERNAL_OPTIMIZED_CPU_CHECK_
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_KERNELS_INTERNAL_OPTIMIZED_CPU_CHECK_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_KERNELS_INTERNAL_OPTIMIZED_CPU_CHECK_H_
|
||||
|
||||
namespace tflite {
|
||||
|
||||
@ -58,4 +58,4 @@ inline bool TestCPUFeatureNeon() { return false; }
|
||||
: Portable##funcname(__VA_ARGS__)
|
||||
#endif
|
||||
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_KERNELS_INTERNAL_OPTIMIZED_CPU_CHECK_
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_KERNELS_INTERNAL_OPTIMIZED_CPU_CHECK_H_
|
||||
|
@ -140,4 +140,4 @@ limitations under the License.
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/src/Tensor/TensorIO.h"
|
||||
|
||||
#include "Eigen/src/Core/util/ReenableStupidWarnings.h"
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_KERNELS_INTERNAL_OPTIMIZED_EIGEN_TENSOR_REDUCED_INSTANTIATIONS_H
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_KERNELS_INTERNAL_OPTIMIZED_EIGEN_TENSOR_REDUCED_INSTANTIATIONS_GOOGLE_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_KERNELS_INTERNAL_MULTITHREAD_CONV
|
||||
#define TENSORFLOW_CONTRIB_LITE_KERNELS_INTERNAL_MULTITHREAD_CONV
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_KERNELS_INTERNAL_OPTIMIZED_MULTITHREADED_CONV_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_KERNELS_INTERNAL_OPTIMIZED_MULTITHREADED_CONV_H_
|
||||
|
||||
#include <assert.h>
|
||||
#include <stdint.h>
|
||||
@ -164,4 +164,4 @@ inline void Conv(const Eigen::ThreadPoolDevice& device, const float* input_data,
|
||||
} // namespace multithreaded_ops
|
||||
} // namespace tflite
|
||||
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_KERNELS_INTERNAL_MULTITHREAD_CONV
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_KERNELS_INTERNAL_OPTIMIZED_MULTITHREADED_CONV_H_
|
||||
|
@ -12,8 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_KERNELS_INTERNAL_OPTIMIZED_OPS_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_KERNELS_INTERNAL_OPTIMIZED_OPS_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_KERNELS_INTERNAL_OPTIMIZED_OPTIMIZED_OPS_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_KERNELS_INTERNAL_OPTIMIZED_OPTIMIZED_OPS_H_
|
||||
|
||||
#include <assert.h>
|
||||
#include <stdint.h>
|
||||
@ -5991,4 +5991,4 @@ inline void TransposeConv(const float* input_data, const Dims<4>& input_dims,
|
||||
#pragma GCC diagnostic pop
|
||||
#endif
|
||||
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_KERNELS_INTERNAL_OPTIMIZED_OPS_H_
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_KERNELS_INTERNAL_OPTIMIZED_OPTIMIZED_OPS_H_
|
||||
|
@ -12,8 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_KERNELS_OP_UTIL_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_KERNELS_OP_UTIL_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_KERNELS_OP_MACROS_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_KERNELS_OP_MACROS_H_
|
||||
|
||||
#include <cstdio>
|
||||
|
||||
@ -31,4 +31,4 @@ limitations under the License.
|
||||
if ((x) != (y)) TF_LITE_FATAL(#x " didn't equal " #y); \
|
||||
} while (0)
|
||||
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_KERNELS_OP_UTIL_H_
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_KERNELS_OP_MACROS_H_
|
||||
|
@ -12,8 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
#ifndef NN_API_SHIM_H0
|
||||
#define NN_API_SHIM_H0
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_NNAPI_NEURALNETWORKSSHIM_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_NNAPI_NEURALNETWORKSSHIM_H_
|
||||
|
||||
#include <dlfcn.h>
|
||||
#include <stdint.h>
|
||||
@ -970,4 +970,4 @@ inline void ANeuralNetworksEvent_free(ANeuralNetworksEvent* event) {
|
||||
|
||||
/**/
|
||||
|
||||
#endif // NN_API_SHIM_H0
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_NNAPI_NEURALNETWORKSSHIM_H_
|
||||
|
@ -14,8 +14,8 @@ limitations under the License.
|
||||
==============================================================================*/
|
||||
// Optional debugging functionality. For small sized binaries, these are not
|
||||
// needed.
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_DEBUG_TOOLS_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_DEBUG_TOOLS_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_OPTIONAL_DEBUG_TOOLS_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_OPTIONAL_DEBUG_TOOLS_H_
|
||||
|
||||
#include "tensorflow/contrib/lite/interpreter.h"
|
||||
|
||||
@ -26,4 +26,4 @@ void PrintInterpreterState(Interpreter* interpreter);
|
||||
|
||||
} // namespace tflite
|
||||
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_DEBUG_TOOLS_H_
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_OPTIONAL_DEBUG_TOOLS_H_
|
||||
|
@ -12,8 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_NNAPI_PARSE_TESTDATA_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_NNAPI_PARSE_TESTDATA_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_TESTING_PARSE_TESTDATA_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_TESTING_PARSE_TESTDATA_H_
|
||||
|
||||
#include <vector>
|
||||
#include "tensorflow/contrib/lite/interpreter.h"
|
||||
@ -72,4 +72,4 @@ bool ParseAndRunTests(std::istream* input, TestRunner* test_runner,
|
||||
} // namespace testing
|
||||
} // namespace tflite
|
||||
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_NNAPI_PARSE_TESTDATA_H_
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_TESTING_PARSE_TESTDATA_H_
|
||||
|
@ -12,8 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_TESTING_TOKENIZER_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_TESTING_TOKENIZER_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_TESTING_TOKENIZE_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_TESTING_TOKENIZE_H_
|
||||
|
||||
#include <istream>
|
||||
#include <string>
|
||||
@ -39,4 +39,4 @@ void Tokenize(std::istream* input, TokenProcessor* processor);
|
||||
} // namespace testing
|
||||
} // namespace tflite
|
||||
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_TESTING_TOKENIZER_H_
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_TESTING_TOKENIZE_H_
|
||||
|
@ -12,8 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_CLUSTER_H
|
||||
#define TENSORFLOW_CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_CLUSTER_H
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_CLUSTER_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_CLUSTER_H_
|
||||
|
||||
#include <string>
|
||||
#include <vector>
|
||||
@ -98,4 +98,4 @@ class ClusterFactoryInterface {
|
||||
|
||||
} // end namespace toco
|
||||
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_CLUSTER_H
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_CLUSTER_H_
|
||||
|
@ -12,8 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_CLUSTERUTILS_H
|
||||
#define TENSORFLOW_CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_CLUSTERUTILS_H
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_CLUSTER_UTILS_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_CLUSTER_UTILS_H_
|
||||
|
||||
#include <string>
|
||||
|
||||
@ -30,4 +30,4 @@ void Transpose2DTensor(const float* tensor, int row, int col,
|
||||
|
||||
} // end namespace toco
|
||||
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_CLUSTERUTILS_H
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_CLUSTER_UTILS_H_
|
||||
|
@ -12,8 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_RESOLVE_CLUSTER_H
|
||||
#define TENSORFLOW_CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_RESOLVE_CLUSTER_H
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_RESOLVE_CLUSTER_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_RESOLVE_CLUSTER_H_
|
||||
|
||||
#include <string>
|
||||
#include <unordered_map>
|
||||
@ -60,4 +60,4 @@ std::unique_ptr<tensorflow::GraphDef> MaybeReplaceCompositeSubgraph(
|
||||
|
||||
} // end namespace toco
|
||||
|
||||
#endif // CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_RESOLVE_CLUSTER_H
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_RESOLVE_CLUSTER_H_
|
||||
|
@ -12,8 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_RESOLVE_SVDF_H
|
||||
#define TENSORFLOW_CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_RESOLVE_SVDF_H
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_RESOLVE_SVDF_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_RESOLVE_SVDF_H_
|
||||
|
||||
#include <string>
|
||||
#include <vector>
|
||||
@ -79,4 +79,4 @@ class SvdfClusterFactory : public ClusterFactoryInterface {
|
||||
|
||||
} // end namespace toco
|
||||
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_RESOLVE_SVDF_H
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_TOCO_TENSORFLOW_GRAPH_MATCHING_RESOLVE_SVDF_H_
|
||||
|
@ -12,8 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_TOCO_TYPES_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_TOCO_TYPES_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_TOCO_TOCO_TYPES_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_TOCO_TOCO_TYPES_H_
|
||||
|
||||
#include <string>
|
||||
#include "tensorflow/core/platform/platform.h"
|
||||
@ -42,4 +42,4 @@ using tensorflow::uint8;
|
||||
|
||||
} // namespace toco
|
||||
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_TOCO_TYPES_H_
|
||||
#endif // TENSORFLOW_CONTRIB_LITE_TOCO_TOCO_TYPES_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_TOOLS_BENCHMARK_MODEL_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_TOOLS_BENCHMARK_MODEL_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_TOOLS_BENCHMARK_BENCHMARK_MODEL_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_TOOLS_BENCHMARK_BENCHMARK_MODEL_H_
|
||||
|
||||
#include <cmath>
|
||||
#include <limits>
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_TOOLS_BENCHMARK_TFLITE_MODEL_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_TOOLS_BENCHMARK_TFLITE_MODEL_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_TOOLS_BENCHMARK_BENCHMARK_TFLITE_MODEL_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_TOOLS_BENCHMARK_BENCHMARK_TFLITE_MODEL_H_
|
||||
|
||||
#include <memory>
|
||||
#include <string>
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_TOOLS_COMMAND_LINE_FLAGS_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_TOOLS_COMMAND_LINE_FLAGS_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_TOOLS_BENCHMARK_COMMAND_LINE_FLAGS_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_TOOLS_BENCHMARK_COMMAND_LINE_FLAGS_H_
|
||||
|
||||
#include <functional>
|
||||
#include <string>
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_TOOLS_LOGGING_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_TOOLS_LOGGING_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_LITE_TOOLS_BENCHMARK_LOGGING_H_
|
||||
#define TENSORFLOW_CONTRIB_LITE_TOOLS_BENCHMARK_LOGGING_H_
|
||||
|
||||
// LOG and CHECK macros for benchmarks.
|
||||
|
||||
|
@ -12,8 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
#ifndef TENSORFLOW_CORE_KERNELS_NCCL_COMMUNICATOR_H_
|
||||
#define TENSORFLOW_CORE_KERNELS_NCCL_COMMUNICATOR_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_NCCL_KERNELS_NCCL_MANAGER_H_
|
||||
#define TENSORFLOW_CONTRIB_NCCL_KERNELS_NCCL_MANAGER_H_
|
||||
|
||||
#ifdef GOOGLE_CUDA
|
||||
|
||||
@ -135,4 +135,4 @@ class NcclManager {
|
||||
|
||||
#endif // GOOGLE_CUDA
|
||||
|
||||
#endif // TENSORFLOW_CORE_KERNELS_NCCL_COMMUNICATOR_H_
|
||||
#endif // TENSORFLOW_CONTRIB_NCCL_KERNELS_NCCL_MANAGER_H_
|
||||
|
@ -14,8 +14,8 @@
|
||||
// limitations under the License.
|
||||
// =============================================================================
|
||||
|
||||
#ifndef TENSORFLOW_KERNELS_PERIODICRESAMPLE_OP_H_
|
||||
#define TENSORFLOW_KERNELS_PERIODICRESAMPLE_OP_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_PERIODIC_RESAMPLE_KERNELS_PERIODIC_RESAMPLE_OP_H_
|
||||
#define TENSORFLOW_CONTRIB_PERIODIC_RESAMPLE_KERNELS_PERIODIC_RESAMPLE_OP_H_
|
||||
|
||||
#include <cmath>
|
||||
#include <type_traits>
|
||||
@ -421,4 +421,4 @@ class PeriodicResampleOpGrad : public tensorflow::OpKernel {
|
||||
tensorflow::PartialTensorShape desired_shape;
|
||||
};
|
||||
|
||||
#endif // TENSORFLOW_KERNELS_PERIODICRESAMPLE_OP_H_
|
||||
#endif // TENSORFLOW_CONTRIB_PERIODIC_RESAMPLE_KERNELS_PERIODIC_RESAMPLE_OP_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_CORE_KERNELS_PARTIAL_REDUCTION_OPS_H_
|
||||
#define TENSORFLOW_CORE_KERNELS_PARTIAL_REDUCTION_OPS_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_REDUCE_SLICE_OPS_KERNELS_REDUCE_SLICE_OPS_H_
|
||||
#define TENSORFLOW_CONTRIB_REDUCE_SLICE_OPS_KERNELS_REDUCE_SLICE_OPS_H_
|
||||
|
||||
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
|
||||
#include "tensorflow/core/framework/tensor.h"
|
||||
@ -81,4 +81,4 @@ CALL_ALL_REDUCEOPS(ReduceSliceFunctorReduceop)
|
||||
} // namespace functor
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_CORE_KERNELS_PARTIAL_REDUCTION_OPS_H_
|
||||
#endif // TENSORFLOW_CONTRIB_REDUCE_SLICE_OPS_KERNELS_REDUCE_SLICE_OPS_H_
|
||||
|
@ -13,8 +13,8 @@
|
||||
// limitations under the License.
|
||||
// =============================================================================
|
||||
|
||||
#ifndef LEARNING_LIB_TENSOR_FOREST_HYBRID_CORE_OPS_UTILS_H_
|
||||
#define LEARNING_LIB_TENSOR_FOREST_HYBRID_CORE_OPS_UTILS_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_TENSOR_FOREST_HYBRID_CORE_OPS_UTILS_H_
|
||||
#define TENSORFLOW_CONTRIB_TENSOR_FOREST_HYBRID_CORE_OPS_UTILS_H_
|
||||
#include <vector>
|
||||
|
||||
#include "tensorflow/core/framework/tensor.h"
|
||||
@ -43,4 +43,4 @@ void GetFeatureSet(int32 tree_num, int32 node_num, int32 random_seed,
|
||||
} // namespace tensorforest
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // LEARNING_LIB_TENSOR_FOREST_HYBRID_CORE_OPS_UTILS_H_
|
||||
#endif // TENSORFLOW_CONTRIB_TENSOR_FOREST_HYBRID_CORE_OPS_UTILS_H_
|
||||
|
@ -15,8 +15,8 @@
|
||||
// This is a surrogate for using a proto, since it doesn't seem to be possible
|
||||
// to use protos in a dynamically-loaded/shared-linkage library, which is
|
||||
// what is used for custom ops in tensorflow/contrib.
|
||||
#ifndef TENSORFLOW_CONTRIB_TENSOR_FOREST_CORE_OPS_DATA_SPEC_H_
|
||||
#define TENSORFLOW_CONTRIB_TENSOR_FOREST_CORE_OPS_DATA_SPEC_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_TENSOR_FOREST_KERNELS_DATA_SPEC_H_
|
||||
#define TENSORFLOW_CONTRIB_TENSOR_FOREST_KERNELS_DATA_SPEC_H_
|
||||
#include <unordered_map>
|
||||
|
||||
#include "tensorflow/core/lib/strings/numbers.h"
|
||||
@ -139,4 +139,4 @@ class TensorForestDataSpec {
|
||||
} // namespace tensorforest
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_CONTRIB_TENSOR_FOREST_CORE_OPS_DATA_SPEC_H_
|
||||
#endif // TENSORFLOW_CONTRIB_TENSOR_FOREST_KERNELS_DATA_SPEC_H_
|
||||
|
@ -12,8 +12,8 @@
|
||||
// See the License for the specific language governing permissions and
|
||||
// limitations under the License.
|
||||
// =============================================================================
|
||||
#ifndef TENSORFLOW_CONTRIB_TENSOR_FOREST_CORE_OPS_TREE_UTILS_H_
|
||||
#define TENSORFLOW_CONTRIB_TENSOR_FOREST_CORE_OPS_TREE_UTILS_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_TENSOR_FOREST_KERNELS_TREE_UTILS_H_
|
||||
#define TENSORFLOW_CONTRIB_TENSOR_FOREST_KERNELS_TREE_UTILS_H_
|
||||
|
||||
#include <limits>
|
||||
|
||||
@ -302,4 +302,4 @@ void GetParentWeightedMean(float leaf_sum, const float* leaf_data,
|
||||
} // namespace tensorforest
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_CONTRIB_TENSOR_FOREST_CORE_OPS_TREE_UTILS_H_
|
||||
#endif // TENSORFLOW_CONTRIB_TENSOR_FOREST_KERNELS_TREE_UTILS_H_
|
||||
|
@ -42,4 +42,4 @@ class TRTResourceManager {
|
||||
} // namespace tensorrt
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_CONTRIB_TENSORRT_RESOURCE_TRT_RESOURCE_MANAGER_H_
|
||||
#endif // TENSORFLOW_CONTRIB_TENSORRT_RESOURCES_TRT_RESOURCE_MANAGER_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_CONTRIB_GRPC_VERBS_CLIENT_H_
|
||||
#define TENSORFLOW_CONTRIB_GRPC_VERBS_CLIENT_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_VERBS_GRPC_VERBS_CLIENT_H_
|
||||
#define TENSORFLOW_CONTRIB_VERBS_GRPC_VERBS_CLIENT_H_
|
||||
|
||||
#include "tensorflow/contrib/verbs/grpc_verbs_service_impl.h"
|
||||
#include "tensorflow/contrib/verbs/verbs_service.pb.h"
|
||||
@ -47,4 +47,4 @@ class GrpcVerbsClient {
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_CONTRIB_GRPC_VERBS_CLIENT_H_
|
||||
#endif // TENSORFLOW_CONTRIB_VERBS_GRPC_VERBS_CLIENT_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_CONTRIB_GRPC_VERBS_SERVICE_IMPL_H_
|
||||
#define TENSORFLOW_CONTRIB_GRPC_VERBS_SERVICE_IMPL_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_VERBS_GRPC_VERBS_SERVICE_IMPL_H_
|
||||
#define TENSORFLOW_CONTRIB_VERBS_GRPC_VERBS_SERVICE_IMPL_H_
|
||||
|
||||
#include "grpcpp/impl/codegen/async_stream.h"
|
||||
#include "grpcpp/impl/codegen/async_unary_call.h"
|
||||
@ -86,4 +86,4 @@ class VerbsService GRPC_FINAL {
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_CONTRIB_GRPC_VERBS_SERVICE_IMPL_H_
|
||||
#endif // TENSORFLOW_CONTRIB_VERBS_GRPC_VERBS_SERVICE_IMPL_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_CONTRIB_RDMA_UTIL_H_
|
||||
#define TENSORFLOW_CONTRIB_RDMA_UTIL_H_
|
||||
#ifndef TENSORFLOW_CONTRIB_VERBS_VERBS_UTIL_H_
|
||||
#define TENSORFLOW_CONTRIB_VERBS_VERBS_UTIL_H_
|
||||
|
||||
#include <string>
|
||||
|
||||
@ -30,4 +30,4 @@ class VerbsUtil {
|
||||
};
|
||||
|
||||
} // namespace tensorflow
|
||||
#endif // TENSORFLOW_CONTRIB_RDMA_UTIL_H_
|
||||
#endif // TENSORFLOW_CONTRIB_VERBS_VERBS_UTIL_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_BFC_ALLOCATOR_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_BFC_ALLOCATOR_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_BFC_ALLOCATOR_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_BFC_ALLOCATOR_H_
|
||||
|
||||
#include <array>
|
||||
#include <memory>
|
||||
@ -451,4 +451,4 @@ class BFCAllocator : public VisitableAllocator {
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_BFC_ALLOCATOR_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_BFC_ALLOCATOR_H_
|
||||
|
@ -12,8 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_BUF_RENDEZVOUS_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_BUF_RENDEZVOUS_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_BUF_RENDEZVOUS_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_BUF_RENDEZVOUS_H_
|
||||
|
||||
#include <functional>
|
||||
#include <string>
|
||||
@ -100,4 +100,4 @@ class BufRendezvous {
|
||||
void PurgeTable(const Status& s, HookTable* table);
|
||||
};
|
||||
} // namespace tensorflow
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_BUF_RENDEZVOUS_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_BUF_RENDEZVOUS_H_
|
||||
|
@ -12,8 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_COLLECTIVE_EXECUTOR_MGR_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_COLLECTIVE_EXECUTOR_MGR_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_COLLECTIVE_EXECUTOR_MGR_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_COLLECTIVE_EXECUTOR_MGR_H_
|
||||
|
||||
#include "tensorflow/core/framework/collective.h"
|
||||
#include "tensorflow/core/lib/gtl/flatmap.h"
|
||||
@ -72,4 +72,4 @@ class CollectiveExecutorMgr : public CollectiveExecutorMgrInterface {
|
||||
};
|
||||
|
||||
} // namespace tensorflow
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_COLLECTIVE_EXECUTOR_MGR_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_COLLECTIVE_EXECUTOR_MGR_H_
|
||||
|
@ -12,8 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_COLLECTIVE_PARAM_RESOLVER_LOCAL_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_COLLECTIVE_PARAM_RESOLVER_LOCAL_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_COLLECTIVE_PARAM_RESOLVER_LOCAL_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_COLLECTIVE_PARAM_RESOLVER_LOCAL_H_
|
||||
|
||||
#include <string>
|
||||
|
||||
@ -237,4 +237,4 @@ class CollectiveParamResolverLocal : public ParamResolverInterface {
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_COLLECTIVE_PARAM_RESOLVER_LOCAL_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_COLLECTIVE_PARAM_RESOLVER_LOCAL_H_
|
||||
|
@ -12,8 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_COLLECTIVE_RMA_LOCAL_ACCESS_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_COLLECTIVE_RMA_LOCAL_ACCESS_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_COLLECTIVE_RMA_LOCAL_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_COLLECTIVE_RMA_LOCAL_H_
|
||||
#include "tensorflow/core/common_runtime/buf_rendezvous.h"
|
||||
#include "tensorflow/core/common_runtime/device_mgr.h"
|
||||
#include "tensorflow/core/framework/collective.h"
|
||||
@ -89,4 +89,4 @@ class CollectiveRemoteAccessLocal : public PerStepCollectiveRemoteAccess {
|
||||
};
|
||||
|
||||
} // namespace tensorflow
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_COLLECTIVE_RMA_LOCAL_ACCESS_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_COLLECTIVE_RMA_LOCAL_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_CONSTANT_FOLDING_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_CONSTANT_FOLDING_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_CONSTANT_FOLDING_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_CONSTANT_FOLDING_H_
|
||||
|
||||
#include "tensorflow/core/common_runtime/device.h"
|
||||
#include "tensorflow/core/framework/function.h"
|
||||
@ -66,4 +66,4 @@ Status ConstantFold(const ConstantFoldingOptions& opts,
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_CONSTANT_FOLDING_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_CONSTANT_FOLDING_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_DEBUGGER_STATE_INTERFACE_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_DEBUGGER_STATE_INTERFACE_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_DEBUGGER_STATE_INTERFACE_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_DEBUGGER_STATE_INTERFACE_H_
|
||||
|
||||
#include <memory>
|
||||
|
||||
@ -117,4 +117,4 @@ class DebugGraphDecoratorRegistry {
|
||||
|
||||
} // end namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_DEBUGGER_STATE_INTERFACE_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_DEBUGGER_STATE_INTERFACE_H_
|
||||
|
@ -26,8 +26,8 @@ limitations under the License.
|
||||
// * Task numbers are within the specified replica, so there are as
|
||||
// many "task zeros" as replicas.
|
||||
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_DEVICE_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_DEVICE_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_DEVICE_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_DEVICE_H_
|
||||
|
||||
#include <memory>
|
||||
#include <string>
|
||||
@ -183,4 +183,4 @@ class Device : public DeviceBase {
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_DEVICE_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_DEVICE_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_DEVICE_FACTORY_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_DEVICE_FACTORY_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_DEVICE_FACTORY_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_DEVICE_FACTORY_H_
|
||||
|
||||
#include <string>
|
||||
#include <vector>
|
||||
@ -126,4 +126,4 @@ class Registrar {
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_DEVICE_FACTORY_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_DEVICE_FACTORY_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_DEVICE_MGR_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_DEVICE_MGR_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_DEVICE_MGR_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_DEVICE_MGR_H_
|
||||
|
||||
#include <string>
|
||||
#include <unordered_map>
|
||||
@ -77,4 +77,4 @@ class DeviceMgr {
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_DEVICE_MGR_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_DEVICE_MGR_H_
|
||||
|
@ -12,8 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_DEVICE_RESOLVER_LOCAL_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_DEVICE_RESOLVER_LOCAL_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_DEVICE_RESOLVER_LOCAL_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_DEVICE_RESOLVER_LOCAL_H_
|
||||
|
||||
#include <string>
|
||||
|
||||
@ -45,4 +45,4 @@ class DeviceResolverLocal : public DeviceResolverInterface {
|
||||
};
|
||||
|
||||
} // namespace tensorflow
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_DEVICE_RESOLVER_LOCAL_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_DEVICE_RESOLVER_LOCAL_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_DEVICE_SET_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_DEVICE_SET_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_DEVICE_SET_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_DEVICE_SET_H_
|
||||
|
||||
#include <memory>
|
||||
#include <unordered_map>
|
||||
@ -86,4 +86,4 @@ class DeviceSet {
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_DEVICE_SET_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_DEVICE_SET_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_DIRECT_SESSION_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_DIRECT_SESSION_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_DIRECT_SESSION_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_DIRECT_SESSION_H_
|
||||
|
||||
#include <atomic>
|
||||
#include <memory>
|
||||
@ -399,4 +399,4 @@ class DirectSession : public Session {
|
||||
|
||||
} // end namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_DIRECT_SESSION_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_DIRECT_SESSION_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_DMA_HELPER_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_DMA_HELPER_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_DMA_HELPER_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_DMA_HELPER_H_
|
||||
|
||||
#include "tensorflow/core/framework/tensor.h"
|
||||
|
||||
@ -35,4 +35,4 @@ class DMAHelper {
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_DMA_HELPER_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_DMA_HELPER_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_C_EAGER_RUNTIME_H_
|
||||
#define TENSORFLOW_C_EAGER_RUNTIME_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_EAGER_ATTR_BUILDER_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_EAGER_ATTR_BUILDER_H_
|
||||
|
||||
// Support for eager execution of TensorFlow kernels.
|
||||
|
||||
@ -154,4 +154,4 @@ AttrBuilder& AttrBuilder::Set(StringPiece attr_name,
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_C_EAGER_RUNTIME_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_EAGER_ATTR_BUILDER_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_EXECUTOR_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_EXECUTOR_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_EXECUTOR_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_EXECUTOR_H_
|
||||
|
||||
#include "tensorflow/core/common_runtime/device.h"
|
||||
#include "tensorflow/core/framework/rendezvous.h"
|
||||
@ -235,4 +235,4 @@ void DeleteNonCachedKernel(OpKernel* kernel);
|
||||
|
||||
} // end namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_EXECUTOR_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_EXECUTOR_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_FUNCTION_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_FUNCTION_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_FUNCTION_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_FUNCTION_H_
|
||||
|
||||
#include <functional>
|
||||
#include <memory>
|
||||
@ -170,4 +170,4 @@ Status FunctionDefToBodyHelper(
|
||||
FunctionBody** fbody);
|
||||
} // end namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_FUNCTION_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_FUNCTION_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_GPU_GPU_BFC_ALLOCATOR_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_GPU_GPU_BFC_ALLOCATOR_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_GPU_GPU_BFC_ALLOCATOR_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_GPU_GPU_BFC_ALLOCATOR_H_
|
||||
|
||||
#include <memory>
|
||||
#include <string>
|
||||
@ -89,4 +89,4 @@ class GPUMemAllocator : public SubAllocator {
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_GPU_GPU_BFC_ALLOCATOR_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_GPU_GPU_BFC_ALLOCATOR_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_GPU_GPU_CUDA_MALLOC_ALLOCATOR_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_GPU_GPU_CUDA_MALLOC_ALLOCATOR_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_GPU_GPU_CUDAMALLOC_ALLOCATOR_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_GPU_GPU_CUDAMALLOC_ALLOCATOR_H_
|
||||
|
||||
#include <memory>
|
||||
|
||||
@ -51,4 +51,4 @@ class GPUcudaMallocAllocator : public VisitableAllocator {
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_GPU_GPU_CUDAMALLOC_ALLOCATOR_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_GPU_GPU_CUDAMALLOC_ALLOCATOR_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_GPU_GPU_DEBUG_ALLOCATOR_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_GPU_GPU_DEBUG_ALLOCATOR_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_GPU_GPU_DEBUG_ALLOCATOR_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_GPU_GPU_DEBUG_ALLOCATOR_H_
|
||||
|
||||
#include <memory>
|
||||
#include <string>
|
||||
@ -88,4 +88,4 @@ class GPUNanResetAllocator : public VisitableAllocator {
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_GPU_GPU_DEBUG_ALLOCATOR_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_GPU_GPU_DEBUG_ALLOCATOR_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_GPU_GPU_EVENT_MGR_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_GPU_GPU_EVENT_MGR_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_GPU_GPU_EVENT_MGR_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_GPU_GPU_EVENT_MGR_H_
|
||||
|
||||
#include <deque>
|
||||
#include <vector>
|
||||
@ -203,4 +203,4 @@ class EventMgr {
|
||||
};
|
||||
|
||||
} // namespace tensorflow
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_GPU_GPU_EVENT_MGR_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_GPU_GPU_EVENT_MGR_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_GPU_GPU_INIT_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_GPU_GPU_INIT_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_GPU_GPU_INIT_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_GPU_GPU_INIT_H_
|
||||
|
||||
#include "tensorflow/core/lib/core/status.h"
|
||||
|
||||
@ -36,4 +36,4 @@ stream_executor::Platform* GPUMachineManager();
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_GPU_GPU_INIT_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_GPU_GPU_INIT_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_GPU_GPU_STREAM_UTIL_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_GPU_GPU_STREAM_UTIL_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_GPU_GPU_STREAM_UTIL_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_GPU_GPU_STREAM_UTIL_H_
|
||||
|
||||
#include <unordered_map>
|
||||
|
||||
@ -42,4 +42,4 @@ Status AssignStreams(const Graph* graph, const AssignStreamsOpts& opts,
|
||||
} // namespace gpu_stream_util
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_GPU_GPU_STREAM_UTIL_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_GPU_GPU_STREAM_UTIL_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_GPU_GPU_UTIL_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_GPU_GPU_UTIL_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_GPU_GPU_UTIL_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_GPU_GPU_UTIL_H_
|
||||
|
||||
#include "tensorflow/core/common_runtime/device.h"
|
||||
#include "tensorflow/core/common_runtime/dma_helper.h"
|
||||
@ -108,4 +108,4 @@ class GPUUtil {
|
||||
};
|
||||
|
||||
} // namespace tensorflow
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_GPU_GPU_UTIL_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_GPU_GPU_UTIL_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_GPU_DEVICE_CONTEXT_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_GPU_DEVICE_CONTEXT_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_GPU_DEVICE_CONTEXT_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_GPU_DEVICE_CONTEXT_H_
|
||||
|
||||
#include "tensorflow/core/common_runtime/device.h"
|
||||
#include "tensorflow/core/framework/device_base.h"
|
||||
@ -75,4 +75,4 @@ class GPUDeviceContext : public DeviceContext {
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_GPU_DEVICE_CONTEXT_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_GPU_DEVICE_CONTEXT_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_KERNEL_BENCHMARK_TESTLIB_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_KERNEL_BENCHMARK_TESTLIB_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_KERNEL_BENCHMARK_TESTLIB_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_KERNEL_BENCHMARK_TESTLIB_H_
|
||||
|
||||
#include <string>
|
||||
#include <vector>
|
||||
@ -65,4 +65,4 @@ class Benchmark {
|
||||
} // end namespace test
|
||||
} // end namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_KERNEL_BENCHMARK_TESTLIB_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_KERNEL_BENCHMARK_TESTLIB_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_LOCAL_DEVICE_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_LOCAL_DEVICE_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_LOCAL_DEVICE_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_LOCAL_DEVICE_H_
|
||||
|
||||
#include "tensorflow/core/common_runtime/device.h"
|
||||
#include "tensorflow/core/framework/device_attributes.pb.h"
|
||||
@ -54,4 +54,4 @@ class LocalDevice : public Device {
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_LOCAL_DEVICE_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_LOCAL_DEVICE_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_PLACER_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_PLACER_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_PLACER_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_PLACER_H_
|
||||
|
||||
#include <string>
|
||||
#include <unordered_map>
|
||||
@ -100,4 +100,4 @@ class Placer {
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_PLACER_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_PLACER_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_RENDEZVOUS_MGR_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_RENDEZVOUS_MGR_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_RENDEZVOUS_MGR_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_RENDEZVOUS_MGR_H_
|
||||
|
||||
#include <string>
|
||||
#include <unordered_map>
|
||||
@ -87,4 +87,4 @@ class IntraProcessRendezvous : public Rendezvous {
|
||||
|
||||
} // end namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_RENDEZVOUS_MGR_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_RENDEZVOUS_MGR_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_SESSION_FACTORY_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_SESSION_FACTORY_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_SESSION_FACTORY_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_SESSION_FACTORY_H_
|
||||
|
||||
#include <string>
|
||||
|
||||
@ -73,4 +73,4 @@ class SessionFactory {
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_SESSION_FACTORY_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_SESSION_FACTORY_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_COMMON_RUNTIME_VISITABLE_ALLOCATOR_H_
|
||||
#define TENSORFLOW_COMMON_RUNTIME_VISITABLE_ALLOCATOR_H_
|
||||
#ifndef TENSORFLOW_CORE_COMMON_RUNTIME_VISITABLE_ALLOCATOR_H_
|
||||
#define TENSORFLOW_CORE_COMMON_RUNTIME_VISITABLE_ALLOCATOR_H_
|
||||
|
||||
#include <functional>
|
||||
#include "tensorflow/core/framework/allocator.h"
|
||||
@ -76,4 +76,4 @@ class TrackingVisitableAllocator : public TrackingAllocator,
|
||||
VisitableAllocator* allocator_;
|
||||
};
|
||||
} // namespace tensorflow
|
||||
#endif // TENSORFLOW_COMMON_RUNTIME_VISITABLE_ALLOCATOR_H_
|
||||
#endif // TENSORFLOW_CORE_COMMON_RUNTIME_VISITABLE_ALLOCATOR_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_DEBUG_CALLBACK_REGISTRY_H_
|
||||
#define TENSORFLOW_DEBUG_CALLBACK_REGISTRY_H_
|
||||
#ifndef TENSORFLOW_CORE_DEBUG_DEBUG_CALLBACK_REGISTRY_H_
|
||||
#define TENSORFLOW_CORE_DEBUG_DEBUG_CALLBACK_REGISTRY_H_
|
||||
|
||||
#include <functional>
|
||||
#include <map>
|
||||
@ -68,4 +68,4 @@ class DebugCallbackRegistry {
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_DEBUG_CALLBACK_REGISTRY_H_
|
||||
#endif // TENSORFLOW_CORE_DEBUG_DEBUG_CALLBACK_REGISTRY_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_DEBUG_NODE_INSERTER_H_
|
||||
#define TENSORFLOW_DEBUG_NODE_INSERTER_H_
|
||||
#ifndef TENSORFLOW_CORE_DEBUG_DEBUG_GRAPH_UTILS_H_
|
||||
#define TENSORFLOW_CORE_DEBUG_DEBUG_GRAPH_UTILS_H_
|
||||
|
||||
#include <unordered_map>
|
||||
#include <vector>
|
||||
@ -123,4 +123,4 @@ class DebugNodeInserter {
|
||||
};
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_DEBUG_NODE_INSERTER_H_
|
||||
#endif // TENSORFLOW_CORE_DEBUG_DEBUG_GRAPH_UTILS_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_DEBUG_GRPC_TESTLIB_H_
|
||||
#define TENSORFLOW_DEBUG_GRPC_TESTLIB_H_
|
||||
#ifndef TENSORFLOW_CORE_DEBUG_DEBUG_GRPC_TESTLIB_H_
|
||||
#define TENSORFLOW_CORE_DEBUG_DEBUG_GRPC_TESTLIB_H_
|
||||
|
||||
#include <atomic>
|
||||
#include <unordered_set>
|
||||
@ -84,4 +84,4 @@ bool PollTillFirstRequestSucceeds(const string& server_url,
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_DEBUG_GRPC_TESTLIB_H_
|
||||
#endif // TENSORFLOW_CORE_DEBUG_DEBUG_GRPC_TESTLIB_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_DEBUG_IO_UTILS_H_
|
||||
#define TENSORFLOW_DEBUG_IO_UTILS_H_
|
||||
#ifndef TENSORFLOW_CORE_DEBUG_DEBUG_IO_UTILS_H_
|
||||
#define TENSORFLOW_CORE_DEBUG_DEBUG_IO_UTILS_H_
|
||||
|
||||
#include <cstddef>
|
||||
#include <functional>
|
||||
@ -398,4 +398,4 @@ class DebugGrpcIO {
|
||||
} // namespace tensorflow
|
||||
#endif // #ifndef(PLATFORM_WINDOWS)
|
||||
|
||||
#endif // TENSORFLOW_DEBUG_IO_UTILS_H_
|
||||
#endif // TENSORFLOW_CORE_DEBUG_DEBUG_IO_UTILS_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_DEBUG_NODE_KEY_H_
|
||||
#define TENSORFLOW_DEBUG_NODE_KEY_H_
|
||||
#ifndef TENSORFLOW_CORE_DEBUG_DEBUG_NODE_KEY_H_
|
||||
#define TENSORFLOW_CORE_DEBUG_DEBUG_NODE_KEY_H_
|
||||
|
||||
#include <string>
|
||||
|
||||
@ -48,4 +48,4 @@ struct DebugNodeKey {
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_DEBUG_NODE_KEY_H_
|
||||
#endif // TENSORFLOW_CORE_DEBUG_DEBUG_NODE_KEY_H_
|
||||
|
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
|
||||
limitations under the License.
|
||||
==============================================================================*/
|
||||
|
||||
#ifndef TENSORFLOW_DEBUGGER_STATE_IMPL_H_
|
||||
#define TENSORFLOW_DEBUGGER_STATE_IMPL_H_
|
||||
#ifndef TENSORFLOW_CORE_DEBUG_DEBUGGER_STATE_IMPL_H_
|
||||
#define TENSORFLOW_CORE_DEBUG_DEBUGGER_STATE_IMPL_H_
|
||||
|
||||
#include "tensorflow/core/common_runtime/debugger_state_interface.h"
|
||||
|
||||
@ -58,4 +58,4 @@ class DebugGraphDecorator : public DebugGraphDecoratorInterface {
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_DEBUGGER_STATE_IMPL_H_
|
||||
#endif // TENSORFLOW_CORE_DEBUG_DEBUGGER_STATE_IMPL_H_
|
||||
|
@ -99,4 +99,4 @@ struct MasterEnv {
|
||||
|
||||
} // end namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW_CORE_DISTRIBUTED_RUNTIME_MASTER_H_
|
||||
#endif // TENSORFLOW_CORE_DISTRIBUTED_RUNTIME_MASTER_ENV_H_
|
||||
|
@ -721,4 +721,4 @@ class NonOwnedProtoRunStepResponse : public MutableRunStepResponseWrapper {
|
||||
|
||||
} // namespace tensorflow
|
||||
|
||||
#endif // TENSORFLOW
|
||||
#endif // TENSORFLOW_CORE_DISTRIBUTED_RUNTIME_MESSAGE_WRAPPERS_H_
|
||||
|
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in New Issue
Block a user