Searched refs:cuda (Results 1 - 25 of 53) sorted by relevance

123

/external/tensorflow/tensorflow/stream_executor/cuda/
H A Dcuda_platform_id.cc16 #include "tensorflow/stream_executor/cuda/cuda_platform_id.h"
20 namespace cuda { namespace in namespace:perftools::gputools
24 } // namespace cuda
H A Dcuda_platform_id.h23 namespace cuda { namespace in namespace:perftools::gputools
25 // Opaque and unique identifier for the cuda platform.
32 } // namespace cuda
H A Dcuda_activation.cc16 #include "tensorflow/stream_executor/cuda/cuda_activation.h"
18 #include "tensorflow/stream_executor/cuda/cuda_driver.h"
24 namespace cuda { namespace in namespace:perftools::gputools
42 } // namespace cuda
H A Dcuda_activation.h21 // code does not also have to depend on cuda.h.
33 namespace cuda { namespace in namespace:perftools::gputools
52 // The cuda.h-using datatype that we wrap.
58 } // namespace cuda
H A Dcuda_event.h19 #include "tensorflow/stream_executor/cuda/cuda_driver.h"
20 #include "tensorflow/stream_executor/cuda/cuda_stream.h"
26 namespace cuda { namespace in namespace:perftools::gputools
60 } // namespace cuda
H A Dcuda_rng.cc16 #include "tensorflow/stream_executor/cuda/cuda_rng.h"
18 #include "tensorflow/stream_executor/cuda/cuda_activation.h"
19 #include "tensorflow/stream_executor/cuda/cuda_gpu_executor.h"
20 #include "tensorflow/stream_executor/cuda/cuda_helpers.h"
21 #include "tensorflow/stream_executor/cuda/cuda_platform_id.h"
22 #include "tensorflow/stream_executor/cuda/cuda_stream.h"
29 #include "cuda/include/curand.h"
59 namespace cuda { namespace in namespace:perftools::gputools
69 cuda::ScopedActivateExecutorContext sac{parent}; \
273 } // namespace cuda
[all...]
H A Dcuda_timer.h23 #include "tensorflow/stream_executor/cuda/cuda_driver.h"
24 #include "tensorflow/stream_executor/cuda/cuda_gpu_executor.h"
28 namespace cuda { namespace in namespace:perftools::gputools
80 } // namespace cuda
H A Dcuda_diagnostics.h27 namespace cuda { namespace in namespace:perftools::gputools
95 } // namespace cuda
H A Dcuda_platform.cc16 #include "tensorflow/stream_executor/cuda/cuda_platform.h"
18 #include "tensorflow/stream_executor/cuda/cuda_driver.h"
19 #include "tensorflow/stream_executor/cuda/cuda_gpu_executor.h"
20 #include "tensorflow/stream_executor/cuda/cuda_platform_id.h"
29 namespace cuda { namespace in namespace:perftools::gputools
138 if (!cuda::CUDADriver::Init().ok()) {
195 } // namespace cuda
201 std::unique_ptr<cuda::CudaPlatform> platform(new cuda::CudaPlatform);
H A Dcuda_stream.cc16 #include "tensorflow/stream_executor/cuda/cuda_stream.h"
18 #include "tensorflow/stream_executor/cuda/cuda_gpu_executor.h"
24 namespace cuda { namespace in namespace:perftools::gputools
61 } // namespace cuda
H A Dcuda_stream.h22 #include "tensorflow/stream_executor/cuda/cuda_driver.h"
28 namespace cuda { namespace in namespace:perftools::gputools
91 } // namespace cuda
H A Dcuda_event.cc16 #include "tensorflow/stream_executor/cuda/cuda_event.h"
18 #include "tensorflow/stream_executor/cuda/cuda_gpu_executor.h"
19 #include "tensorflow/stream_executor/cuda/cuda_stream.h"
24 namespace cuda { namespace in namespace:perftools::gputools
70 } // namespace cuda
H A Dcuda_helpers.h27 #include "cuda/include/cuComplex.h"
28 #include "cuda/include/cuda.h"
36 namespace cuda { namespace in namespace:perftools::gputools
103 } // namespace cuda
H A Dcuda_kernel.h27 #include "tensorflow/stream_executor/cuda/cuda_driver.h"
31 #include "cuda/include/cuda.h"
45 namespace cuda { namespace in namespace:perftools::gputools
126 } // namespace cuda
H A Dcuda_platform.h36 namespace cuda { namespace in namespace:perftools::gputools
106 } // namespace cuda
H A Dcuda_rng.h34 namespace cuda { namespace in namespace:perftools::gputools
100 } // namespace cuda
H A Dcuda_timer.cc16 #include "tensorflow/stream_executor/cuda/cuda_timer.h"
18 #include "tensorflow/stream_executor/cuda/cuda_driver.h"
19 #include "tensorflow/stream_executor/cuda/cuda_gpu_executor.h"
20 #include "tensorflow/stream_executor/cuda/cuda_stream.h"
25 namespace cuda { namespace in namespace:perftools::gputools
86 } // namespace cuda
H A Dcuda_fft.cc16 #include "tensorflow/stream_executor/cuda/cuda_fft.h"
20 #include "tensorflow/stream_executor/cuda/cuda_activation.h"
21 #include "tensorflow/stream_executor/cuda/cuda_gpu_executor.h"
22 #include "tensorflow/stream_executor/cuda/cuda_helpers.h"
23 #include "tensorflow/stream_executor/cuda/cuda_platform_id.h"
24 #include "tensorflow/stream_executor/cuda/cuda_stream.h"
36 namespace cuda { namespace in namespace:perftools::gputools
51 cuda::ScopedActivateExecutorContext sac{parent}; \
543 } // namespace cuda
553 gpu::cuda
[all...]
/external/tensorflow/tensorflow/contrib/makefile/
H A Dbuild_all_android.sh62 if [ ! -d ${JETPACK}/cuda ]; then
63 ln -s $(ls -d ${JETPACK}/cuda-*/|sort -r|head -n1) ${JETPACK}/cuda
65 if [ ! -d ${JETPACK}/cuda ]; then
66 ln -s $(ls -d ${JETPACK}/cuda-*/|sort -r|head -n1) ${JETPACK}/cuda
/external/tensorflow/tensorflow/tools/ci_build/windows/gpu/bazel/
H A Drun_cc_test_windows.sh59 # bazel test --config=win-cuda -k $slow_compiling_test --test_output=errors
60 bazel test -c opt --config=win-cuda -k $passing_tests --test_output=errors
/external/tensorflow/tensorflow/tools/ci_build/linux/gpu/
H A Drun_cc_core.sh37 bazel test --config=cuda --test_tag_filters=-no_oss,-oss_serial,-no_gpu,-benchmark-test -k \
H A Drun_py3_core.sh37 bazel test --config=cuda --test_tag_filters=-no_oss,-oss_serial,-no_gpu,-benchmark-test -k \
/external/tensorflow/tensorflow/tools/ci_build/xla/linux/gpu/
H A Drun_py3.sh38 bazel test --config=cuda --test_tag_filters=-no_gpu,-benchmark-test -k \
/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_cudamalloc_allocator.cc17 #include "cuda/include/cuda.h"
18 #include "tensorflow/stream_executor/cuda/cuda_activation.h"
41 gpu::cuda::ScopedActivateExecutorContext scoped_activation{stream_exec_};
/external/tensorflow/tools/
H A Dtf_env_collect.sh108 echo '== cuda libs ==================================================='
111 find /usr/local -type f -name 'libcudart*' 2>/dev/null | grep cuda | grep -v "\\.cache" >> ${OUTPUT_FILE}
112 find /usr/local -type f -name 'libudnn*' 2>/dev/null | grep cuda | grep -v "\\.cache" >> ${OUTPUT_FILE}

Completed in 232 milliseconds

123