Home
last modified time | relevance | path

Searched refs:dim_z (Results 1 – 3 of 3) sorted by relevance

/external/eigen/unsupported/test/
Dcxx11_tensor_reduction_cuda.cu63 int dim_z = 128; in test_first_dim_reductions() local
65 Tensor<Type, 3, DataLayout> in(dim_x, dim_y, dim_z); in test_first_dim_reductions()
77 Type* in_data = (Type*)dev.allocate(dim_x*dim_y*dim_z*sizeof(Type)); in test_first_dim_reductions()
78 Type* out_data = (Type*)dev.allocate(dim_z*dim_y*sizeof(Type)); in test_first_dim_reductions()
79 Eigen::TensorMap<Eigen::Tensor<Type, 3, DataLayout> > gpu_in(in_data, dim_x, dim_y, dim_z); in test_first_dim_reductions()
80 Eigen::TensorMap<Eigen::Tensor<Type, 2, DataLayout> > gpu_out(out_data, dim_y, dim_z); in test_first_dim_reductions()
86 Tensor<Type, 2, DataLayout> redux_gpu(dim_y, dim_z); in test_first_dim_reductions()
103 int dim_z = 33; in test_last_dim_reductions() local
105 Tensor<Type, 3, DataLayout> in(dim_x, dim_y, dim_z); in test_last_dim_reductions()
117 Type* in_data = (Type*)dev.allocate(dim_x*dim_y*dim_z*sizeof(Type)); in test_last_dim_reductions()
[all …]
Dcxx11_tensor_reduction_sycl.cpp59 int dim_z = 67; in test_first_dim_reductions_sycl() local
61 array<int, 3> tensorRange = {{dim_x, dim_y, dim_z}}; in test_first_dim_reductions_sycl()
64 array<int, 2> reduced_tensorRange = {{dim_y, dim_z}}; in test_first_dim_reductions_sycl()
97 int dim_z = 47; in test_last_dim_reductions_sycl() local
99 array<int, 3> tensorRange = {{dim_x, dim_y, dim_z}}; in test_last_dim_reductions_sycl()
/external/llvm-project/lldb/source/Plugins/LanguageRuntime/RenderScript/RenderScriptRuntime/
DRenderScriptRuntime.cpp2120 uint32_t dim_z = alloc->dimension.get()->dim_3; in JITAllocationSize() local
2130 if (dim_z == 0) in JITAllocationSize()
2131 dim_z = 1; in JITAllocationSize()
2133 alloc->size = dim_x * dim_y * dim_z * *alloc->element.datum_size.get(); in JITAllocationSize()
2146 dim_z = dim_z == 0 ? 0 : dim_z - 1; in JITAllocationSize()
2149 *alloc->address.get(), dim_x, dim_y, dim_z); in JITAllocationSize()
3261 uint32_t dim_z = alloc->dimension.get()->dim_3; in DumpAllocation() local
3262 dim_z = dim_z == 0 ? 1 : dim_z; in DumpAllocation()
3275 for (uint32_t z = 0; z < dim_z; ++z) { in DumpAllocation()