Home
last modified time | relevance | path

Searched defs:gpu_device (Results 1 – 25 of 29) sorted by relevance

12

/external/eigen/unsupported/test/
Dcxx11_tensor_gpu.cu40 Eigen::GpuDevice gpu_device(&stream); in test_gpu_nullary() local
91 Eigen::GpuDevice gpu_device(&stream); in test_gpu_elementwise_small() local
146 Eigen::GpuDevice gpu_device(&stream); in test_gpu_elementwise() local
188 Eigen::GpuDevice gpu_device(&stream); in test_gpu_props() local
226 Eigen::GpuDevice gpu_device(&stream); in test_gpu_reduction() local
286 Eigen::GpuDevice gpu_device(&stream); in test_gpu_contraction() local
343 Eigen::GpuDevice gpu_device(&stream); in test_gpu_convolution_1d() local
396 Eigen::GpuDevice gpu_device(&stream); in test_gpu_convolution_inner_dim_col_major_1d() local
449 Eigen::GpuDevice gpu_device(&stream); in test_gpu_convolution_inner_dim_row_major_1d() local
503 Eigen::GpuDevice gpu_device(&stream); in test_gpu_convolution_2d() local
[all …]
Dcxx11_tensor_of_float16_gpu.cu25 Eigen::GpuDevice gpu_device(&stream); in test_gpu_numext() local
65 Eigen::GpuDevice gpu_device(&stream); in test_gpu_conversion() local
100 Eigen::GpuDevice gpu_device(&stream); in test_gpu_unary() local
137 Eigen::GpuDevice gpu_device(&stream); in test_gpu_elementwise() local
179 Eigen::GpuDevice gpu_device(&stream); in test_gpu_trancendental() local
273 Eigen::GpuDevice gpu_device(&stream); in test_gpu_contractions() local
328 Eigen::GpuDevice gpu_device(&stream); in test_gpu_reductions() local
380 Eigen::GpuDevice gpu_device(&stream); in test_gpu_full_reductions() local
425 Eigen::GpuDevice gpu_device(&stream); in test_gpu_forced_evals() local
Dcxx11_tensor_complex_gpu.cu38 Eigen::GpuDevice gpu_device(&stream); in test_cuda_nullary() local
74 Eigen::GpuDevice gpu_device(&stream); in test_cuda_sum_reductions() local
110 Eigen::GpuDevice gpu_device(&stream); in test_cuda_mean_reductions() local
146 Eigen::GpuDevice gpu_device(&stream); in test_cuda_product_reductions() local
Dcxx11_tensor_random_gpu.cu32 Eigen::GpuDevice gpu_device(&stream); in test_gpu_random_uniform() local
57 Eigen::GpuDevice gpu_device(&stream); in test_gpu_random_normal() local
Dcxx11_tensor_argmax_gpu.cu46 Eigen::GpuDevice gpu_device(&stream); in test_gpu_simple_argmax() local
108 Eigen::GpuDevice gpu_device(&stream); in test_gpu_argmax_dim() local
197 Eigen::GpuDevice gpu_device(&stream); in test_gpu_argmin_dim() local
Dcxx11_tensor_contract_gpu.cu57 Eigen::GpuDevice gpu_device(&stream); in test_gpu_contraction() local
121 Eigen::GpuDevice gpu_device(&stream); in test_scalar() local
Dcxx11_tensor_scan_gpu.cu46 Eigen::GpuDevice gpu_device(&stream); in test_gpu_cumsum() local
Dcxx11_tensor_cast_float16_gpu.cu23 Eigen::GpuDevice gpu_device(&stream); in test_gpu_conversion() local
Dcxx11_tensor_complex_cwise_ops_gpu.cu32 Eigen::GpuDevice gpu_device(&stream); in test_cuda_complex_cwise_ops() local
Dcxx11_tensor_reduction_gpu.cu23 Eigen::GpuDevice gpu_device(&stream); in test_full_reductions() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_placer_test.cc32 DeviceProperties gpu_device; in TEST() local
62 DeviceProperties gpu_device; in TEST() local
121 DeviceProperties gpu_device; in TEST() local
178 DeviceProperties gpu_device; in TEST() local
195 DeviceProperties gpu_device; in TEST() local
290 DeviceProperties gpu_device; in TEST() local
Danalytical_cost_estimator_test.cc38 DeviceProperties gpu_device; in SetUp() local
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_util.cc266 void GPUUtil::CopyGPUTensorToCPU(Device* gpu_device, in CopyGPUTensorToCPU()
313 Device* gpu_device, Tensor* gpu_tensor, in CopyCPUTensorToGPU()
392 Status GPUUtil::Sync(Device* gpu_device) { in Sync()
401 Status GPUUtil::SyncAll(Device* gpu_device) { in SyncAll()
439 uint64 GPUUtil::Checksum(Device* gpu_device, in Checksum()
468 void GPUUtil::CopyGPUTensorToSameGPU(Device* gpu_device, in CopyGPUTensorToSameGPU()
/external/tensorflow/tensorflow/core/kernels/
Dconcat_lib_gpu_impl.cu.cc117 const Eigen::GpuDevice& gpu_device, in ConcatGPUSlice()
138 void ConcatGPUImpl(const Eigen::GpuDevice& gpu_device, in ConcatGPUImpl()
Ddynamic_stitch_op_gpu.cu.cc52 void DynamicStitchGPUImpl(const Eigen::GpuDevice& gpu_device, in DynamicStitchGPUImpl()
Dsplit_lib_gpu.cu.cc210 const Eigen::GpuDevice& gpu_device, bool fixed_size, const T* input_ptr, in Run()
/external/tensorflow/tensorflow/core/runtime_fallback/kernel/gpu/
Dconversion_function.cc51 auto gpu_device = static_cast<BaseGPUDevice*>(expected_device.get()); in ConvertKernelFallbackTensorOnGpuToDenseGpuTensor() local
/external/tensorflow/tensorflow/core/runtime_fallback/util/gpu/
Dgpu_utils.cc36 auto gpu_device = static_cast<tensorflow::BaseGPUDevice*>(device); in GetTfrtGpuPlatformHelper() local
/external/tensorflow/tensorflow/core/tfrt/runtime/
Druntime.cc111 auto gpu_device = tfrt::gpu::GetOrCreateGpuDevice( in InitializeOpHandlers() local
/external/tensorflow/tensorflow/compiler/xla/python/
Dxla.cc306 py::class_<GpuDevice, PjRtDevice, ClientAndPtr<GpuDevice>> gpu_device( in PYBIND11_MODULE() local
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmemory_optimizer_test.cc213 DeviceProperties gpu_device; in CreateVirtualCluster() local
Dgeneric_layout_optimizer_test.cc181 DeviceProperties gpu_device; in SetUp() local
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition_test.cc62 const char gpu_device[] = "/job:a/replica:0/task:0/device:GPU:0"; variable
/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
Druntime_fallback_kernels.cc227 auto gpu_device = static_cast<tensorflow::BaseGPUDevice*>(device); in InjectTfGpuResourcesHelper() local
/external/tensorflow/tensorflow/c/
Dc_api_test.cc1086 const string gpu_device = GPUDeviceName(); in TEST() local
1094 const string gpu_device = GPUDeviceName(); in TEST() local

12