| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gpu_compiler.h | 102 int cc_major, int cc_minor);
104 // The compilation_cache_ map is a cache from {ptx string, cc_major, cc_minor}
117 CompilationCacheKey(std::string ptx, int cc_major, int cc_minor) argument
118 : ptx(std::move(ptx)), cc_major(cc_major), cc_minor(cc_minor) {}
121 int cc_minor; member in struct:xla::gpu::CompilationCacheKey
127 key.cc_minor);
133 return a.cc_major == b.cc_major && a.cc_minor == b.cc_minor &&
|
| H A D | gpu_compiler.cc | 401 int cc_minor) {
436 tensorflow::strings::StrCat("-arch=sm_", cc_major, cc_minor)};
600 int cc_major, cc_minor; local
602 &cc_minor)) {
606 cc_minor = 0;
612 TF_ASSIGN_OR_RETURN(ptx, CompileToPtx(&llvm_module, {cc_major, cc_minor},
648 CompilePtxOrGetCachedResult(ptx, cc_major, cc_minor);
668 ptx, cubin, {cc_major, cc_minor}, std::move(thunk_schedule),
680 int cc_minor) {
694 std::forward_as_tuple(ptx, cc_major, cc_minor),
400 CompilePtx(const string& ptx, int cc_major, int cc_minor) argument
678 CompilePtxOrGetCachedResult(const string& ptx, int cc_major, int cc_minor) argument
[all...] |
| /external/tensorflow/tensorflow/stream_executor/ |
| H A D | dnn.cc | 26 bool with_winograd_nonfused, int cc_major, int cc_minor,
32 bool with_winograd_nonfused, int cc_major, int cc_minor,
38 bool with_winograd_nonfused, int cc_major, int cc_minor,
25 GetConvolveAlgorithms( bool with_winograd_nonfused, int cc_major, int cc_minor, std::vector<AlgorithmDesc>* out_algorithms) argument
31 GetConvolveBackwardDataAlgorithms( bool with_winograd_nonfused, int cc_major, int cc_minor, std::vector<AlgorithmDesc>* out_algorithms) argument
37 GetConvolveBackwardFilterAlgorithms( bool with_winograd_nonfused, int cc_major, int cc_minor, std::vector<AlgorithmDesc>* out_algorithms) argument
|
| H A D | stream_executor_pimpl.cc | 302 int cc_major, cc_minor; local
303 GetDeviceDescription().cuda_compute_capability(&cc_major, &cc_minor);
305 cc_minor, out_algorithms);
315 int cc_major, cc_minor; local
316 GetDeviceDescription().cuda_compute_capability(&cc_major, &cc_minor);
318 with_winograd_nonfused, cc_major, cc_minor, out_algorithms);
328 int cc_major, cc_minor; local
329 GetDeviceDescription().cuda_compute_capability(&cc_major, &cc_minor);
331 with_winograd_nonfused, cc_major, cc_minor, out_algorithms);
|
| /external/tensorflow/tensorflow/contrib/fused_conv/kernels/ |
| H A D | fused_conv2d_bias_activation_op.cc | 325 int cc_major, cc_minor; local
327 &cc_minor);
329 ctx, ((cc_major == 6 && cc_minor >= 1) || cc_major > 6),
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| H A D | gpu_device.cc | 991 int cc_minor; local
992 if (!desc.cuda_compute_capability(&cc_major, &cc_minor)) {
994 cc_minor = 0;
1000 ", compute capability: ", cc_major, ".", cc_minor);
1325 int cc_minor; local
1326 if (!description.cuda_compute_capability(&cc_major, &cc_minor)) {
1329 cc_minor = 0;
1333 << " minor: " << cc_minor
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| H A D | cuda_blas.cc | 1852 int cc_major, cc_minor; local
1854 &cc_minor);
2161 int cc_major, cc_minor;
2163 &cc_major, &cc_minor) &&
|
| H A D | cuda_driver.cc | 1383 int *cc_minor,
1386 *cc_minor = 0;
1387 CUresult result = cuDeviceComputeCapability(cc_major, cc_minor, device);
1382 GetComputeCapability(int *cc_major, int *cc_minor, CUdevice device) argument
|
| H A D | cuda_gpu_executor.cc | 922 int cc_minor; member in struct:perftools::gputools::cuda::UnqueryableDeviceParams
1120 if (params.cc_major == cc_major_ && params.cc_minor == cc_minor_) {
|
| H A D | cuda_dnn.cc | 2639 bool with_winograd_nonfused, int cc_major, int cc_minor,
2673 bool with_winograd_nonfused, int cc_major, int cc_minor,
2703 bool with_winograd_nonfused, int cc_major, int cc_minor,
3053 int cc_major, cc_minor;
3055 &cc_minor);
3056 if (cc_major < 6 || (cc_major == 6 && cc_minor < 1)) {
2638 GetConvolveAlgorithms( bool with_winograd_nonfused, int cc_major, int cc_minor, std::vector<dnn::AlgorithmDesc>* out_algorithms) argument
2672 GetConvolveBackwardDataAlgorithms( bool with_winograd_nonfused, int cc_major, int cc_minor, std::vector<dnn::AlgorithmDesc>* out_algorithms) argument
2702 GetConvolveBackwardFilterAlgorithms( bool with_winograd_nonfused, int cc_major, int cc_minor, std::vector<dnn::AlgorithmDesc>* out_algorithms) argument
|