Execute TODO to rename io.* to save_restore_tensor.*. This will
hopefully reduce confusion since io.* is not the implementation of the ".../kernels:io" build target. Change: 115593814
This commit is contained in:
		
							parent
							
								
									ad3ef4c05b
								
							
						
					
					
						commit
						eec5477ab6
					
				@ -405,11 +405,11 @@ cc_library(
 | 
			
		||||
        "//tensorflow/core/kernels:conv_2d.h",
 | 
			
		||||
        "//tensorflow/core/kernels:fill_functor.h",
 | 
			
		||||
        "//tensorflow/core/kernels:initializable_lookup_table.h",
 | 
			
		||||
        "//tensorflow/core/kernels:io.h",
 | 
			
		||||
        "//tensorflow/core/kernels:lookup_util.h",
 | 
			
		||||
        "//tensorflow/core/kernels:ops_util.h",
 | 
			
		||||
        "//tensorflow/core/kernels:queue_base.h",
 | 
			
		||||
        "//tensorflow/core/kernels:reader_base.h",
 | 
			
		||||
        "//tensorflow/core/kernels:save_restore_tensor.h",
 | 
			
		||||
        "//tensorflow/core/kernels:typed_queue.h",
 | 
			
		||||
    ],
 | 
			
		||||
    deprecation = "Use libraries from .../kernels instead.",
 | 
			
		||||
 | 
			
		||||
@ -170,9 +170,8 @@ cc_library(
 | 
			
		||||
 | 
			
		||||
cc_library(
 | 
			
		||||
    name = "save_restore_tensor",
 | 
			
		||||
    # TODO(josh11b): Rename io.* -> save_restore_tensor.*
 | 
			
		||||
    srcs = ["io.cc"],
 | 
			
		||||
    hdrs = ["io.h"],
 | 
			
		||||
    srcs = ["save_restore_tensor.cc"],
 | 
			
		||||
    hdrs = ["save_restore_tensor.h"],
 | 
			
		||||
    deps = [
 | 
			
		||||
        "//tensorflow/core:framework",
 | 
			
		||||
        "//tensorflow/core:lib",
 | 
			
		||||
@ -1031,11 +1030,11 @@ filegroup(
 | 
			
		||||
        "avgpooling_op.h",
 | 
			
		||||
        "control_flow_ops.h",
 | 
			
		||||
        "conv_2d.h",
 | 
			
		||||
        "io.h",
 | 
			
		||||
        "maxpooling_op.h",
 | 
			
		||||
        "reduction_ops.h",
 | 
			
		||||
        "reduction_ops_common.h",
 | 
			
		||||
        "relu_op.h",
 | 
			
		||||
        "save_restore_tensor.h",
 | 
			
		||||
        "softplus_op.h",
 | 
			
		||||
        "softsign_op.h",
 | 
			
		||||
        "tile_ops.h",
 | 
			
		||||
@ -1083,7 +1082,6 @@ filegroup(
 | 
			
		||||
    srcs = [
 | 
			
		||||
        "dynamic_stitch_op.cc",
 | 
			
		||||
        "in_topk_op.cc",
 | 
			
		||||
        "io.cc",
 | 
			
		||||
        "lrn_op.cc",
 | 
			
		||||
        "maxpooling_op.cc",
 | 
			
		||||
        "reduction_ops_common.cc",
 | 
			
		||||
@ -1095,6 +1093,7 @@ filegroup(
 | 
			
		||||
        "relu_op.cc",
 | 
			
		||||
        "restore_op.cc",
 | 
			
		||||
        "save_op.cc",
 | 
			
		||||
        "save_restore_tensor.cc",
 | 
			
		||||
        "softplus_op.cc",
 | 
			
		||||
        "softsign_op.cc",
 | 
			
		||||
        "sparse_to_dense_op.cc",
 | 
			
		||||
 | 
			
		||||
@ -14,7 +14,7 @@ limitations under the License.
 | 
			
		||||
==============================================================================*/
 | 
			
		||||
 | 
			
		||||
// See docs in ../ops/io_ops.cc.
 | 
			
		||||
#include "tensorflow/core/kernels/io.h"
 | 
			
		||||
#include "tensorflow/core/kernels/save_restore_tensor.h"
 | 
			
		||||
 | 
			
		||||
#include "tensorflow/core/framework/op_kernel.h"
 | 
			
		||||
#include "tensorflow/core/lib/core/errors.h"
 | 
			
		||||
 | 
			
		||||
@ -14,7 +14,7 @@ limitations under the License.
 | 
			
		||||
==============================================================================*/
 | 
			
		||||
 | 
			
		||||
// See docs in ../ops/io_ops.cc
 | 
			
		||||
#include "tensorflow/core/kernels/io.h"
 | 
			
		||||
#include "tensorflow/core/kernels/save_restore_tensor.h"
 | 
			
		||||
 | 
			
		||||
#include "tensorflow/core/framework/op_kernel.h"
 | 
			
		||||
#include "tensorflow/core/framework/types.h"
 | 
			
		||||
 | 
			
		||||
@ -13,11 +13,10 @@ See the License for the specific language governing permissions and
 | 
			
		||||
limitations under the License.
 | 
			
		||||
==============================================================================*/
 | 
			
		||||
 | 
			
		||||
// See docs in ../ops/io_ops.cc
 | 
			
		||||
#include <unordered_map>
 | 
			
		||||
 | 
			
		||||
#include <vector>
 | 
			
		||||
#include "tensorflow/core/kernels/io.h"
 | 
			
		||||
#include "tensorflow/core/kernels/save_restore_tensor.h"
 | 
			
		||||
 | 
			
		||||
#include "tensorflow/core/framework/op_kernel.h"
 | 
			
		||||
#include "tensorflow/core/framework/register_types.h"
 | 
			
		||||
@ -13,8 +13,8 @@ See the License for the specific language governing permissions and
 | 
			
		||||
limitations under the License.
 | 
			
		||||
==============================================================================*/
 | 
			
		||||
 | 
			
		||||
#ifndef TENSORFLOW_KERNELS_IO_H_
 | 
			
		||||
#define TENSORFLOW_KERNELS_IO_H_
 | 
			
		||||
#ifndef TENSORFLOW_KERNELS_SAVE_RESTORE_TENSOR_H_
 | 
			
		||||
#define TENSORFLOW_KERNELS_SAVE_RESTORE_TENSOR_H_
 | 
			
		||||
 | 
			
		||||
#include "tensorflow/core/util/tensor_slice_reader.h"
 | 
			
		||||
#include "tensorflow/core/util/tensor_slice_writer.h"
 | 
			
		||||
@ -50,4 +50,4 @@ void RestoreTensor(OpKernelContext* context,
 | 
			
		||||
 | 
			
		||||
}  // namespace tensorflow
 | 
			
		||||
 | 
			
		||||
#endif  // TENSORFLOW_KERNELS_IO_H_
 | 
			
		||||
#endif  // TENSORFLOW_KERNELS_SAVE_RESTORE_TENSOR_H_
 | 
			
		||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user