提交 eec5477a 编写于 作者: J Josh Levenberg 提交者: TensorFlower Gardener

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
上级 ad3ef4c0
......@@ -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_
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册