• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Copyright 2018 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"""Ignite File System for checkpointing and communication with TensorBoard.
16
17Apache Ignite is a memory-centric distributed database, caching, and
18processing platform for transactional, analytical, and streaming workloads,
19delivering in-memory speeds at petabyte scale. In addition to database
20functionality Apache Ignite provides a distributed file system called
21IGFS (https://ignite.apache.org/features/igfs.html). IGFS delivers a similar
22functionality to Hadoop HDFS, but only in-memory. In fact, in addition to
23its own APIs, IGFS implements Hadoop FileSystem API and can be transparently
24plugged into Hadoop or Spark deployments. This contrib package contains an
25integration between IGFS and TensorFlow.
26"""
27
28from __future__ import absolute_import
29from __future__ import division
30from __future__ import print_function
31
32import os
33
34from tensorflow.contrib.ignite.python.ops import ignite_op_loader  # pylint: disable=unused-import
35from tensorflow.python.framework import load_library
36from tensorflow.python.platform import resource_loader
37
38file_system_library = os.path.join(resource_loader.get_data_files_path(),
39                                   "../../_ignite_ops.so")
40load_library.load_file_system_library(file_system_library)
41