experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #24295 from ROCmSoftwarePlatform:upstream-staging-rename-gpu-device-array

Browse Source 
PiperOrigin-RevId: 238321615
This commit is contained in:
TensorFlower Gardener 2019-03-13 15:45:45 -07:00
commit dd7daa8cc7
13 changed files with 22 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
tensorflow/core/kernels/BUILD
View File

@ -232,8 +232,8 @@ tf_kernel_library(
"concat_lib_gpu_impl.cu.cc",
"concat_lib.h",
"concat_lib_gpu.h",
"cuda_device_array.h",
"cuda_device_array_gpu.h",
"gpu_device_array.h",
"gpu_device_array_gpu.h",
],
deps = [
":bounds_check",
@ -632,7 +632,7 @@ tf_kernel_library(
"split_lib_gpu.h",
],
deps = [
":cuda_device_array",
":gpu_device_array",
"//tensorflow/core:framework",
"//third_party/eigen3",
],
@ -684,10 +684,10 @@ cc_header_only_library(
)
cc_library(
name = "cuda_device_array",
name = "gpu_device_array",
hdrs = [
"cuda_device_array.h",
"cuda_device_array_gpu.h",
"gpu_device_array.h",
"gpu_device_array_gpu.h",
],
visibility = ["//tensorflow:__subpackages__"],
deps = [
@ -1118,14 +1118,14 @@ tf_kernel_library(
tf_kernel_library(
name = "split_op",
gpu_srcs = ["cuda_device_array.h"],
gpu_srcs = ["gpu_device_array.h"],
prefix = "split_op",
deps = ARRAY_DEPS + [":split_lib"],
)
tf_kernel_library(
name = "split_v_op",
gpu_srcs = ["cuda_device_array.h"],
gpu_srcs = ["gpu_device_array.h"],
prefix = "split_v_op",
deps = ARRAY_DEPS + [":split_lib"],
)
@ -2160,8 +2160,8 @@ tf_kernel_library(
tf_kernel_library(
name = "dynamic_stitch_op",
gpu_srcs = [
"cuda_device_array.h",
"cuda_device_array_gpu.h",
"gpu_device_array.h",
"gpu_device_array_gpu.h",
],
prefix = "dynamic_stitch_op",
deps = DYNAMIC_DEPS,
@ -3309,7 +3309,7 @@ tf_kernel_library(
tf_kernel_library(
name = "bucketize_op",
gpu_srcs = ["cuda_device_array.h"],
gpu_srcs = ["gpu_device_array.h"],
prefix = "bucketize_op",
deps = ARRAY_DEPS,
)

2
tensorflow/core/kernels/bucketize_op_gpu.cu.cc
View File

@ -24,7 +24,7 @@ limitations under the License.
#include "tensorflow/core/framework/tensor.h"
#include "tensorflow/core/framework/tensor_shape.h"
#include "tensorflow/core/kernels/bucketize_op.h"
#include "tensorflow/core/kernels/cuda_device_array.h"
#include "tensorflow/core/kernels/gpu_device_array.h"
#include "tensorflow/core/platform/logging.h"
#include "tensorflow/core/platform/types.h"
#include "tensorflow/core/util/cuda_kernel_helper.h"

2
tensorflow/core/kernels/concat_lib_gpu.cc
View File

@ -27,7 +27,7 @@ limitations under the License.
#if GOOGLE_CUDA
#include "tensorflow/core/kernels/concat_lib_gpu.h"
#include "tensorflow/core/kernels/cuda_device_array.h"
#include "tensorflow/core/kernels/gpu_device_array.h"
namespace tensorflow {
namespace {

2
tensorflow/core/kernels/concat_lib_gpu.h
View File

@ -25,7 +25,7 @@ limitations under the License.
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
#include "tensorflow/core/framework/register_types.h"
#include "tensorflow/core/kernels/concat_lib.h"
#include "tensorflow/core/kernels/cuda_device_array_gpu.h"
#include "tensorflow/core/kernels/gpu_device_array_gpu.h"
namespace tensorflow {

2
tensorflow/core/kernels/concat_lib_gpu_impl.cu.cc
View File

@ -24,7 +24,7 @@ limitations under the License.
#include "tensorflow/core/framework/register_types.h"
#include "tensorflow/core/framework/tensor_types.h"
#include "tensorflow/core/kernels/concat_lib_gpu.h"
#include "tensorflow/core/kernels/cuda_device_array_gpu.h"
#include "tensorflow/core/kernels/gpu_device_array_gpu.h"
#include "tensorflow/core/util/cuda_kernel_helper.h"
namespace tensorflow {

2
tensorflow/core/kernels/dynamic_stitch_op.cc
View File

@ -22,7 +22,7 @@ limitations under the License.
#include "tensorflow/core/lib/core/threadpool.h"
#ifdef GOOGLE_CUDA
#include "tensorflow/core/kernels/cuda_device_array.h"
#include "tensorflow/core/kernels/gpu_device_array.h"
#endif // GOOGLE_CUDA
namespace tensorflow {

2
tensorflow/core/kernels/dynamic_stitch_op_gpu.cu.cc
View File

@ -19,7 +19,7 @@ limitations under the License.
#include "tensorflow/core/framework/register_types.h"
#include "tensorflow/core/framework/tensor_types.h"
#include "tensorflow/core/kernels/cuda_device_array_gpu.h"
#include "tensorflow/core/kernels/gpu_device_array_gpu.h"
#include "tensorflow/core/util/cuda_kernel_helper.h"
namespace tensorflow {

2
tensorflow/core/kernels/cuda_device_array.h → tensorflow/core/kernels/gpu_device_array.h
View File

@ -19,7 +19,7 @@ limitations under the License.
#include "tensorflow/core/common_runtime/gpu/gpu_event_mgr.h"
#include "tensorflow/core/framework/op_kernel.h"
#include "tensorflow/core/kernels/cuda_device_array_gpu.h"
#include "tensorflow/core/kernels/gpu_device_array_gpu.h"
namespace tensorflow {

0
tensorflow/core/kernels/cuda_device_array_gpu.h → tensorflow/core/kernels/gpu_device_array_gpu.h
View File

2
tensorflow/core/kernels/split_lib_gpu.cu.cc
View File

@ -23,7 +23,7 @@ limitations under the License.
#include "tensorflow/core/framework/register_types.h"
#include "tensorflow/core/framework/tensor_types.h"
#include "tensorflow/core/kernels/cuda_device_array_gpu.h"
#include "tensorflow/core/kernels/gpu_device_array_gpu.h"
#include "tensorflow/core/kernels/split_lib_gpu.h"
#include "tensorflow/core/util/cuda_kernel_helper.h"

2
tensorflow/core/kernels/split_lib_gpu.h
View File

@ -24,7 +24,7 @@ limitations under the License.
#include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor"
#include "tensorflow/core/framework/register_types.h"
#include "tensorflow/core/kernels/cuda_device_array_gpu.h"
#include "tensorflow/core/kernels/gpu_device_array_gpu.h"
#include "tensorflow/core/kernels/split_lib.h"
namespace tensorflow {

2
tensorflow/core/kernels/split_op.cc
View File

@ -29,7 +29,7 @@ limitations under the License.
#include "tensorflow/core/util/work_sharder.h"
#if GOOGLE_CUDA
#include "tensorflow/core/common_runtime/gpu/gpu_event_mgr.h"
#include "tensorflow/core/kernels/cuda_device_array.h"
#include "tensorflow/core/kernels/gpu_device_array.h"
#include "tensorflow/core/kernels/split_lib_gpu.h"
#include "tensorflow/core/platform/stream_executor.h"
#endif // GOOGLE_CUDA

2
tensorflow/core/kernels/split_v_op.cc
View File

@ -35,7 +35,7 @@ limitations under the License.
#include "tensorflow/core/util/work_sharder.h"
#if GOOGLE_CUDA
#include "tensorflow/core/common_runtime/gpu/gpu_event_mgr.h"
#include "tensorflow/core/kernels/cuda_device_array.h"
#include "tensorflow/core/kernels/gpu_device_array.h"
#include "tensorflow/core/kernels/split_lib_gpu.h"
#include "tensorflow/core/platform/stream_executor.h"
#endif // GOOGLE_CUDA