Home
last modified time | relevance | path

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

/external/eigen/unsupported/test/
Dcxx11_tensor_cuda.cu39 Eigen::GpuDevice gpu_device(&stream); in test_cuda_nullary() local
90 Eigen::GpuDevice gpu_device(&stream); in test_cuda_elementwise_small() local
145 Eigen::GpuDevice gpu_device(&stream); in test_cuda_elementwise() local
187 Eigen::GpuDevice gpu_device(&stream); in test_cuda_props() local
225 Eigen::GpuDevice gpu_device(&stream); in test_cuda_reduction() local
285 Eigen::GpuDevice gpu_device(&stream); in test_cuda_contraction() local
342 Eigen::GpuDevice gpu_device(&stream); in test_cuda_convolution_1d() local
395 Eigen::GpuDevice gpu_device(&stream); in test_cuda_convolution_inner_dim_col_major_1d() local
448 Eigen::GpuDevice gpu_device(&stream); in test_cuda_convolution_inner_dim_row_major_1d() local
502 Eigen::GpuDevice gpu_device(&stream); in test_cuda_convolution_2d() local
[all …]
Dcxx11_tensor_of_float16_cuda.cu27 Eigen::GpuDevice gpu_device(&stream); in test_cuda_numext() local
67 Eigen::GpuDevice gpu_device(&stream); in test_cuda_conversion() local
102 Eigen::GpuDevice gpu_device(&stream); in test_cuda_unary() local
139 Eigen::GpuDevice gpu_device(&stream); in test_cuda_elementwise() local
181 Eigen::GpuDevice gpu_device(&stream); in test_cuda_trancendental() local
269 Eigen::GpuDevice gpu_device(&stream); in test_cuda_contractions() local
324 Eigen::GpuDevice gpu_device(&stream); in test_cuda_reductions() local
381 Eigen::GpuDevice gpu_device(&stream); in test_cuda_full_reductions() local
431 Eigen::GpuDevice gpu_device(&stream); in test_cuda_forced_evals() local
Dcxx11_tensor_complex_cuda.cu41 Eigen::GpuDevice gpu_device(&stream); in test_cuda_nullary() local
77 Eigen::GpuDevice gpu_device(&stream); in test_cuda_sum_reductions() local
114 Eigen::GpuDevice gpu_device(&stream); in test_cuda_product_reductions() local
Dcxx11_tensor_random_cuda.cu34 Eigen::GpuDevice gpu_device(&stream); in test_cuda_random_uniform() local
59 Eigen::GpuDevice gpu_device(&stream); in test_cuda_random_normal() local
Dcxx11_tensor_argmax_cuda.cu47 Eigen::GpuDevice gpu_device(&stream); in test_cuda_simple_argmax() local
109 Eigen::GpuDevice gpu_device(&stream); in test_cuda_argmax_dim() local
198 Eigen::GpuDevice gpu_device(&stream); in test_cuda_argmin_dim() local
Dcxx11_tensor_contract_cuda.cu58 Eigen::GpuDevice gpu_device(&stream); in test_cuda_contraction() local
122 Eigen::GpuDevice gpu_device(&stream); in test_scalar() local
Dcxx11_tensor_scan_cuda.cu47 Eigen::GpuDevice gpu_device(&stream); in test_cuda_cumsum() local
Dcxx11_tensor_complex_cwise_ops_cuda.cu35 Eigen::GpuDevice gpu_device(&stream); in test_cuda_complex_cwise_ops() local
Dcxx11_tensor_cast_float16_cuda.cu26 Eigen::GpuDevice gpu_device(&stream); in test_cuda_conversion() local
Dcxx11_tensor_reduction_cuda.cu26 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.cc254 void GPUUtil::CopyGPUTensorToCPU(Device* gpu_device, in CopyGPUTensorToCPU()
301 Device* gpu_device, Tensor* gpu_tensor, in CopyCPUTensorToGPU()
346 Status GPUUtil::Sync(Device* gpu_device) { in Sync()
355 Status GPUUtil::SyncAll(Device* gpu_device) { in SyncAll()
393 uint64 GPUUtil::Checksum(Device* gpu_device, in Checksum()
422 void GPUUtil::CopyGPUTensorToSameGPU(Device* gpu_device, in CopyGPUTensorToSameGPU()
Dgpu_device.cc1452 std::unique_ptr<BaseGPUDevice> gpu_device = CreateGPUDevice( in CreateGPUDevice() local
/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/grappler/optimizers/
Dmemory_optimizer_test.cc213 DeviceProperties gpu_device; in CreateVirtualCluster() local
Dgeneric_layout_optimizer_test.cc181 DeviceProperties gpu_device; in SetUp() local
Dgeneric_layout_optimizer_transposer_test.cc362 DeviceProperties gpu_device; in SetUp() local
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition_test.cc61 const char gpu_device[] = "/job:a/replica:0/task:0/device:GPU:0"; variable
/external/tensorflow/tensorflow/c/
Dc_api_test.cc1083 const string gpu_device = GPUDeviceName(); in TEST() local
1091 const string gpu_device = GPUDeviceName(); in TEST() local
/external/tensorflow/tensorflow/c/eager/
Dc_api_test.cc427 auto gpu_device = gpu_arg->device(); in TensorHandleSilentCopy() local
/external/tensorflow/tensorflow/core/common_runtime/
Dplacer_test.cc1563 const string gpu_device = "/job:b/replica:0/task:0/device:FakeGPU:0"; in TEST_F() local