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:conv_2d.h",
|
||||||
"//tensorflow/core/kernels:fill_functor.h",
|
"//tensorflow/core/kernels:fill_functor.h",
|
||||||
"//tensorflow/core/kernels:initializable_lookup_table.h",
|
"//tensorflow/core/kernels:initializable_lookup_table.h",
|
||||||
"//tensorflow/core/kernels:io.h",
|
|
||||||
"//tensorflow/core/kernels:lookup_util.h",
|
"//tensorflow/core/kernels:lookup_util.h",
|
||||||
"//tensorflow/core/kernels:ops_util.h",
|
"//tensorflow/core/kernels:ops_util.h",
|
||||||
"//tensorflow/core/kernels:queue_base.h",
|
"//tensorflow/core/kernels:queue_base.h",
|
||||||
"//tensorflow/core/kernels:reader_base.h",
|
"//tensorflow/core/kernels:reader_base.h",
|
||||||
|
"//tensorflow/core/kernels:save_restore_tensor.h",
|
||||||
"//tensorflow/core/kernels:typed_queue.h",
|
"//tensorflow/core/kernels:typed_queue.h",
|
||||||
],
|
],
|
||||||
deprecation = "Use libraries from .../kernels instead.",
|
deprecation = "Use libraries from .../kernels instead.",
|
||||||
|
@ -170,9 +170,8 @@ cc_library(
|
|||||||
|
|
||||||
cc_library(
|
cc_library(
|
||||||
name = "save_restore_tensor",
|
name = "save_restore_tensor",
|
||||||
# TODO(josh11b): Rename io.* -> save_restore_tensor.*
|
srcs = ["save_restore_tensor.cc"],
|
||||||
srcs = ["io.cc"],
|
hdrs = ["save_restore_tensor.h"],
|
||||||
hdrs = ["io.h"],
|
|
||||||
deps = [
|
deps = [
|
||||||
"//tensorflow/core:framework",
|
"//tensorflow/core:framework",
|
||||||
"//tensorflow/core:lib",
|
"//tensorflow/core:lib",
|
||||||
@ -1031,11 +1030,11 @@ filegroup(
|
|||||||
"avgpooling_op.h",
|
"avgpooling_op.h",
|
||||||
"control_flow_ops.h",
|
"control_flow_ops.h",
|
||||||
"conv_2d.h",
|
"conv_2d.h",
|
||||||
"io.h",
|
|
||||||
"maxpooling_op.h",
|
"maxpooling_op.h",
|
||||||
"reduction_ops.h",
|
"reduction_ops.h",
|
||||||
"reduction_ops_common.h",
|
"reduction_ops_common.h",
|
||||||
"relu_op.h",
|
"relu_op.h",
|
||||||
|
"save_restore_tensor.h",
|
||||||
"softplus_op.h",
|
"softplus_op.h",
|
||||||
"softsign_op.h",
|
"softsign_op.h",
|
||||||
"tile_ops.h",
|
"tile_ops.h",
|
||||||
@ -1083,7 +1082,6 @@ filegroup(
|
|||||||
srcs = [
|
srcs = [
|
||||||
"dynamic_stitch_op.cc",
|
"dynamic_stitch_op.cc",
|
||||||
"in_topk_op.cc",
|
"in_topk_op.cc",
|
||||||
"io.cc",
|
|
||||||
"lrn_op.cc",
|
"lrn_op.cc",
|
||||||
"maxpooling_op.cc",
|
"maxpooling_op.cc",
|
||||||
"reduction_ops_common.cc",
|
"reduction_ops_common.cc",
|
||||||
@ -1095,6 +1093,7 @@ filegroup(
|
|||||||
"relu_op.cc",
|
"relu_op.cc",
|
||||||
"restore_op.cc",
|
"restore_op.cc",
|
||||||
"save_op.cc",
|
"save_op.cc",
|
||||||
|
"save_restore_tensor.cc",
|
||||||
"softplus_op.cc",
|
"softplus_op.cc",
|
||||||
"softsign_op.cc",
|
"softsign_op.cc",
|
||||||
"sparse_to_dense_op.cc",
|
"sparse_to_dense_op.cc",
|
||||||
|
@ -14,7 +14,7 @@ limitations under the License.
|
|||||||
==============================================================================*/
|
==============================================================================*/
|
||||||
|
|
||||||
// See docs in ../ops/io_ops.cc.
|
// 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/op_kernel.h"
|
||||||
#include "tensorflow/core/lib/core/errors.h"
|
#include "tensorflow/core/lib/core/errors.h"
|
||||||
|
@ -14,7 +14,7 @@ limitations under the License.
|
|||||||
==============================================================================*/
|
==============================================================================*/
|
||||||
|
|
||||||
// See docs in ../ops/io_ops.cc
|
// 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/op_kernel.h"
|
||||||
#include "tensorflow/core/framework/types.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.
|
limitations under the License.
|
||||||
==============================================================================*/
|
==============================================================================*/
|
||||||
|
|
||||||
// See docs in ../ops/io_ops.cc
|
|
||||||
#include <unordered_map>
|
#include <unordered_map>
|
||||||
|
|
||||||
#include <vector>
|
#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/op_kernel.h"
|
||||||
#include "tensorflow/core/framework/register_types.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.
|
limitations under the License.
|
||||||
==============================================================================*/
|
==============================================================================*/
|
||||||
|
|
||||||
#ifndef TENSORFLOW_KERNELS_IO_H_
|
#ifndef TENSORFLOW_KERNELS_SAVE_RESTORE_TENSOR_H_
|
||||||
#define TENSORFLOW_KERNELS_IO_H_
|
#define TENSORFLOW_KERNELS_SAVE_RESTORE_TENSOR_H_
|
||||||
|
|
||||||
#include "tensorflow/core/util/tensor_slice_reader.h"
|
#include "tensorflow/core/util/tensor_slice_reader.h"
|
||||||
#include "tensorflow/core/util/tensor_slice_writer.h"
|
#include "tensorflow/core/util/tensor_slice_writer.h"
|
||||||
@ -50,4 +50,4 @@ void RestoreTensor(OpKernelContext* context,
|
|||||||
|
|
||||||
} // namespace tensorflow
|
} // namespace tensorflow
|
||||||
|
|
||||||
#endif // TENSORFLOW_KERNELS_IO_H_
|
#endif // TENSORFLOW_KERNELS_SAVE_RESTORE_TENSOR_H_
|
Loading…
x
Reference in New Issue
Block a user