HomeSort by relevance Sort by last modified time
    Searched refs:cuda (Results 51 - 61 of 61) sorted by null

1 23

  /external/tensorflow/tensorflow/tools/ci_build/
ci_sanity.sh 544 bazel build --action_env=TF_NEED_CUDA=1 --define framework_shared_object=true --config=cuda --nobuild_tests_only tensorflow/core/platform/default/build_config:libtensorflow_cuda_check_deps
548 SANITY_STEPS_DESC=("Python 2 pylint" "Python 3 pylint" "Check that python files have certain __future__ imports" "buildifier check" "bazel nobuild" "pip: license check for external dependencies" "C library: license check for external dependencies" "Java Native Library: license check for external dependencies" "Pip Smoke Test: Checking py_test dependencies exist in pip package" "Check load py_test: Check that BUILD files with py_test target properly load py_test" "Code Link Check: Check there are no broken links" "Check file names for cases" "Check gpu libtensorflow_framework.so does not depend on cuda shared libraries.")
  /external/tensorflow/tensorflow/core/kernels/
where_op.cc 43 #include "tensorflow/core/platform/cuda.h"
45 using stream_executor::cuda::ScopedActivateExecutorContext;
crop_and_resize_op.cc 39 #include "tensorflow/core/platform/cuda.h"
42 using stream_executor::cuda::ScopedActivateExecutorContext;
    [all...]
cuda_solvers.cc 24 #include "cuda/include/cublas_v2.h"
25 #include "cuda/include/cusolverDn.h"
33 #include "tensorflow/core/platform/cuda.h"
38 // The CUDA cublas_api.h API contains const-correctness errors. Instead of
81 using se::cuda::ScopedActivateExecutorContext;
90 // A set of initialized handles to the underlying Cuda libraries used by
121 // cuda stream.
160 // Previously unseen Cuda stream. Initialize a set of Cuda solver library
321 // TODO(rmlarsen): Investigate if the locking is still needed in Cuda 9
    [all...]
segment_reduction_ops.cc 43 #include "tensorflow/core/platform/cuda.h"
45 using stream_executor::cuda::ScopedActivateExecutorContext;
    [all...]
  /external/tensorflow/tensorflow/stream_executor/cuda/
cuda_driver.cc 16 #include "tensorflow/stream_executor/cuda/cuda_driver.h"
28 #include "cuda/include/cuda_runtime_api.h"
29 #include "tensorflow/stream_executor/cuda/cuda_diagnostics.h"
47 // Debugging: on each push and pop of a cuda context, verify the current context
60 // CUDA-runtime-created contexts are avoided, if triple angle
122 // Returns the current context and checks that it is in the set of CUDA contexts
123 // created by StreamExecutor (to ensure that the CUDA runtime didn't create a
126 CUcontext current = cuda::CurrentContextOrDie();
131 << "; a CUDA runtime call "
137 // CUDA driver routines may require a large amount of stack (particularl
1632 namespace cuda { namespace in namespace:stream_executor
    [all...]
cuda_dnn.cc 16 #include "tensorflow/stream_executor/cuda/cuda_dnn.h"
27 #include "tensorflow/stream_executor/cuda/cuda_activation.h"
28 #include "tensorflow/stream_executor/cuda/cuda_diagnostics.h"
29 #include "tensorflow/stream_executor/cuda/cuda_driver.h"
30 #include "tensorflow/stream_executor/cuda/cuda_gpu_executor.h"
31 #include "tensorflow/stream_executor/cuda/cuda_platform_id.h"
32 #include "tensorflow/stream_executor/cuda/cuda_stream.h"
33 #include "tensorflow/stream_executor/cuda/cuda_timer.h"
34 #include "tensorflow/stream_executor/cuda/cudnn_version.h"
47 #include "cuda/include/cudnn.h
    [all...]
cuda_blas.cc 16 #include "cuda/include/cublas_v2.h"
17 #include "cuda/include/cuda.h"
21 #include "tensorflow/stream_executor/cuda/cuda_blas.h"
23 // Both Eigen Half.h and CUDA cuda_fp16.h provide similar typedef for __half. As
36 // NVCC, we choose the first approach for CUDA < 9.0. For CUDA >= 9.0, we have
38 // by CUDA > 9.0 is `__x` while Eigen expects it to be `x`.
43 #include "cuda/include/cuda_fp16.h"
54 #include "tensorflow/stream_executor/cuda/cuda_activation.h
    [all...]
  /external/tensorflow/tensorflow/contrib/makefile/
Makefile 418 NVCC := $(JETPACK)/cuda/bin/nvcc
442 -I$(JETPACK)/cuda/include \
445 -I$(JETPACK)/cuda/extras/CUPTI/include
467 -L$(JETPACK)/cuda/targets/aarch64-linux-androideabi/lib \
468 -L$(JETPACK)/cuda/targets/aarch64-linux-androideabi/lib/stubs \
469 -L$(JETPACK)/cuda/targets/aarch64-linux-androideabi/lib64 \
470 -L$(JETPACK)/cuda/targets/aarch64-linux-androideabi/lib64/stubs \
471 -L$(JETPACK)/cuDNN/aarch64/cuda/lib64 \
689 $(wildcard tensorflow/core/platform/cuda.h) \
  /external/tensorflow/tensorflow/core/nccl/
nccl_manager.cc 22 #include "tensorflow/core/platform/cuda.h"
43 using se::cuda::ScopedActivateExecutorContext;
  /external/tensorflow/tensorflow/core/common_runtime/gpu/
gpu_device.cc 63 #include "tensorflow/core/platform/cuda.h"
79 #include "cuda/cuda_config.h"
90 using se::cuda::ScopedActivateExecutorContext;
107 // CUDA or ROCm memory allocator, and may not work correctly due to race
228 // object (and therefore the same CUDA streams). This is necessary since there
    [all...]

Completed in 1205 milliseconds

1 23