Searched defs:gpu_kernel (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_convolution_sycl.cpp | 69 …Eigen::TensorMap<Eigen::Tensor<DataType, 1, DataLayout, IndexType> > gpu_kernel(d_kernel, kernel_d… in test_larg_expr1D() local
130 …Eigen::TensorMap<Eigen::Tensor<DataType, 2, DataLayout, IndexType> > gpu_kernel(d_kernel, kernel_d… in test_larg_expr2D() local
191 …Eigen::TensorMap<Eigen::Tensor<DataType, 3, DataLayout, IndexType> > gpu_kernel(d_kernel, kernel_d… in test_larg_expr3D() local
244 …Eigen::TensorMap<Eigen::Tensor<DataType, 1, DataLayout, IndexType> > gpu_kernel(d_kernel, kernel_d… in test_evals() local
290 …Eigen::TensorMap<Eigen::Tensor<DataType, 2, DataLayout,IndexType> > gpu_kernel(d_kernel, kernel_di… in test_expr() local
350 …Eigen::TensorMap<Eigen::Tensor<DataType, 1, DataLayout,IndexType> > gpu_kernel(d_kernel, kernel_di… in test_modes() local
431 …Eigen::TensorMap<Eigen::Tensor<DataType, 1, DataLayout,IndexType> > gpu_kernel(d_kernel, kernel_di… in test_strides() local
|
| H A D | cxx11_tensor_gpu.cu | 346 Eigen::TensorMap<Eigen::Tensor<float, 1, DataLayout> > gpu_kernel(d_kernel, 4); in test_gpu_convolution_1d() local
399 Eigen::TensorMap<Eigen::Tensor<float, 1, ColMajor> > gpu_kernel(d_kernel,4); in test_gpu_convolution_inner_dim_col_major_1d() local
452 Eigen::TensorMap<Eigen::Tensor<float, 1, RowMajor> > gpu_kernel(d_kernel, 4); in test_gpu_convolution_inner_dim_row_major_1d() local
506 Eigen::TensorMap<Eigen::Tensor<float, 2, DataLayout> > gpu_kernel(d_kernel,3,4); in test_gpu_convolution_2d() local
570 Eigen::TensorMap<Eigen::Tensor<float, 3, DataLayout> > gpu_kernel(d_kernel,3,4,2); in test_gpu_convolution_3d() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/ |
| H A D | kernel_creator.cc | 319 gpu_module.walk([&](mlir::gpu::GPUFuncOp gpu_kernel) { in LowerKernelBodiesToLowLevelIr()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Loops.cuh | 99 void gpu_kernel(TensorIteratorBase& iter, const func_t& f) { in gpu_kernel() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/profiler/utils/ |
| H A D | group_events_test.cc | 389 const XEvent& gpu_kernel = device_plane->lines(0).events(0); in TEST() local
|