Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/eigen/unsupported/test/
H A 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 …]
H A 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
H A 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
H A 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
H A 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
H A 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
H A Dcxx11_tensor_scan_gpu.cu46 Eigen::GpuDevice gpu_device(&stream); in test_gpu_cumsum() local
H A Dcxx11_tensor_cast_float16_gpu.cu23 Eigen::GpuDevice gpu_device(&stream); in test_gpu_conversion() local
H A Dcxx11_tensor_complex_cwise_ops_gpu.cu32 Eigen::GpuDevice gpu_device(&stream); in test_cuda_complex_cwise_ops() local
H A Dcxx11_tensor_reduction_gpu.cu23 Eigen::GpuDevice gpu_device(&stream); in test_full_reductions() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A 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
H A Danalytical_cost_estimator_test.cc38 DeviceProperties gpu_device; in SetUp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dconcat_lib_gpu_impl.cu.cc117 const Eigen::GpuDevice& gpu_device, in ConcatGPUSlice()
138 void ConcatGPUImpl(const Eigen::GpuDevice& gpu_device, in ConcatGPUImpl()
H A Ddynamic_stitch_op_gpu.cu.cc52 void DynamicStitchGPUImpl(const Eigen::GpuDevice& gpu_device, in DynamicStitchGPUImpl()
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/kernel/gpu/
H A Dconversion_function.cc51 auto gpu_device = static_cast<BaseGPUDevice*>(expected_device.get()); in ConvertKernelFallbackTensorOnGpuToDenseGpuTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/util/gpu/
H A Dgpu_utils.cc36 auto gpu_device = static_cast<tensorflow::BaseGPUDevice*>(device); in GetTfrtGpuPlatformHelper() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/runtime/
H A Druntime.cc111 auto gpu_device = tfrt::gpu::GetOrCreateGpuDevice( in InitializeOpHandlers() local
/aosp_15_r20/external/mesa3d/src/gfxstream/guest/platform/fuchsia/
H A DFuchsiaVirtGpuDevice.cpp106 VirtGpuDevice* gpu_device = nullptr; in osCreateVirtGpuDevice() local
/aosp_15_r20/system/sepolicy/prebuilts/api/29.0/public/
H A Ddevice.te22 type gpu_device, dev_type, mlstrustedobject; type
/aosp_15_r20/system/sepolicy/prebuilts/api/30.0/public/
H A Ddevice.te23 type gpu_device, dev_type, mlstrustedobject; type
/aosp_15_r20/system/sepolicy/prebuilts/api/31.0/public/
H A Ddevice.te25 type gpu_device, dev_type, mlstrustedobject; type
/aosp_15_r20/system/sepolicy/prebuilts/api/32.0/public/
H A Ddevice.te25 type gpu_device, dev_type, mlstrustedobject; type
/aosp_15_r20/external/mesa3d/src/gallium/drivers/radeonsi/ci/
H A Dradeonsi-run-tests.py203 gpu_device = available_gpus[args.gpu][1] variable
/aosp_15_r20/system/sepolicy/prebuilts/api/33.0/public/
H A Ddevice.te25 type gpu_device, dev_type, mlstrustedobject; type

12