Home
last modified time | relevance | path

Searched defs:gpu_out (Results 1 – 17 of 17) sorted by relevance

/aosp_15_r20/external/eigen/unsupported/test/
H A Dcxx11_tensor_gpu.cu97 Eigen::TensorMap<Eigen::Tensor<float, 1>, Eigen::Aligned> gpu_out( in test_gpu_elementwise_small() local
151 …Eigen::TensorMap<Eigen::Tensor<float, 3> > gpu_out(d_out, Eigen::array<Eigen::DenseIndex, 3>(72,53… in test_gpu_elementwise() local
192 Eigen::TensorMap<Eigen::Tensor<bool, 1>, Eigen::Aligned> gpu_out( in test_gpu_props() local
229 Eigen::TensorMap<Eigen::Tensor<float, 2> > gpu_out(d_out, 72,97); in test_gpu_reduction() local
347 Eigen::TensorMap<Eigen::Tensor<float, 4, DataLayout> > gpu_out(d_out, 74,34,11,137); in test_gpu_convolution_1d() local
400 Eigen::TensorMap<Eigen::Tensor<float, 4, ColMajor> > gpu_out(d_out,71,9,11,7); in test_gpu_convolution_inner_dim_col_major_1d() local
453 Eigen::TensorMap<Eigen::Tensor<float, 4, RowMajor> > gpu_out(d_out, 7,9,11,71); in test_gpu_convolution_inner_dim_row_major_1d() local
507 Eigen::TensorMap<Eigen::Tensor<float, 4, DataLayout> > gpu_out(d_out,74,35,8,137); in test_gpu_convolution_2d() local
571 Eigen::TensorMap<Eigen::Tensor<float, 5, DataLayout> > gpu_out(d_out,74,35,8,136,17); in test_gpu_convolution_3d() local
645 Eigen::TensorMap<Eigen::Tensor<Scalar, 2> > gpu_out(d_out, 72, 97); in test_gpu_lgamma() local
[all …]
H A Dcxx11_tensor_builtins_sycl.cpp69 TensorMap<Tensor<DataType, 3, DataLayout, int64_t>> gpu_out(gpu_data_out, tensor_range); in test_unary_builtins_for_scalar() local
93 TensorMap<Tensor<DataType, 3, DataLayout, int64_t>> gpu_out(gpu_data_out, tensor_range); in test_unary_builtins_for_scalar() local
176 TensorMap<Tensor<bool, 3, DataLayout, int64_t>> gpu_out(gpu_data_out, tensor_range); in test_unary_builtins_return_bool() local
234 TensorMap<Tensor<DataType, 3, DataLayout, int64_t>> gpu_out(gpu_data_out, tensor_range); in test_binary_builtins_func() local
265 TensorMap<Tensor<DataType, 3, DataLayout, int64_t>> gpu_out(gpu_data_out, tensor_range); in test_binary_builtins_fixed_arg2() local
H A Dcxx11_tensor_random_gpu.cu34 Eigen::TensorMap<Eigen::Tensor<float, 2> > gpu_out(d_out, 72,97); in test_gpu_random_uniform() local
59 Eigen::TensorMap<Eigen::Tensor<float, 2> > gpu_out(d_out, 72,97); in test_gpu_random_normal() local
H A Dcxx11_tensor_random_sycl.cpp36 TensorMap<Tensor<DataType, 2, DataLayout, IndexType>> gpu_out(d_out, tensorRange); in test_sycl_random_uniform() local
65 TensorMap<Tensor<DataType, 2, DataLayout, IndexType>> gpu_out(d_out, tensorRange); in test_sycl_random_normal() local
H A Dcxx11_tensor_reduction_gpu.cu77 Eigen::TensorMap<Eigen::Tensor<Type, 2, DataLayout> > gpu_out(out_data, dim_y, dim_z); in test_first_dim_reductions() local
117 Eigen::TensorMap<Eigen::Tensor<Type, 2, DataLayout> > gpu_out(out_data, dim_x, dim_y); in test_last_dim_reductions() local
H A Dcxx11_tensor_broadcast_sycl.cpp60 TensorMap<Tensor<DataType, 4, DataLayout, IndexType>> gpu_out(gpu_out_data, out_range); in test_broadcast_sycl_fixed() local
111 TensorMap<Tensor<DataType, 4, DataLayout, IndexType>> gpu_out(gpu_out_data, out_range); in test_broadcast_sycl() local
H A Dcxx11_tensor_argmax_gpu.cu111 Eigen::TensorMap<Eigen::Tensor<DenseIndex, 3, DataLayout>, Aligned > gpu_out(d_out, out_shape); in test_gpu_argmax_dim() local
200 Eigen::TensorMap<Eigen::Tensor<DenseIndex, 3, DataLayout>, Aligned > gpu_out(d_out, out_shape); in test_gpu_argmin_dim() local
H A Dcxx11_tensor_custom_op_sycl.cpp63 TensorType gpu_out(gpu_out_data, tensorResultRange); in test_custom_unary_op_sycl() local
129 TensorType gpu_out(gpu_out_data, tensorResultRange); in test_custom_binary_op_sycl() local
H A Dcxx11_tensor_argmax_sycl.cpp111 … Eigen::TensorMap<Eigen::Tensor<DenseIndex, 3, DataLayout, DenseIndex> > gpu_out(d_out, out_shape); in test_sycl_argmax_dim() local
198 … Eigen::TensorMap<Eigen::Tensor<DenseIndex, 3, DataLayout, DenseIndex> > gpu_out(d_out, out_shape); in test_sycl_argmin_dim() local
H A Dcxx11_tensor_sycl.cpp202 TensorMap<Tensor<DataType, 3, DataLayout, IndexType>> gpu_out(gpu_out_data, tensorRange); in test_sycl_computations() local
329 TensorMap<Tensor<Scalar2, 1, DataLayout, IndexType>> gpu_out(gpu_out_data, tensorRange); in test_sycl_cast() local
H A Dcxx11_tensor_complex_cwise_ops_gpu.cu38 Eigen::TensorMap<Eigen::Tensor<std::complex<T>, 1, 0, int>, Eigen::Aligned> gpu_out( in test_cuda_complex_cwise_ops() local
H A Dcxx11_tensor_forced_eval_sycl.cpp45 …Eigen::TensorMap<Eigen::Tensor<DataType, 3, DataLayout, IndexType>> gpu_out(gpu_out_data, tensorRa… in test_forced_eval_sycl() local
H A Dcxx11_tensor_concatenation_sycl.cpp146 …Eigen::TensorMap<Eigen::Tensor<DataType, 2, DataLayout, IndexType>> gpu_out(gpu_out_data, resRange… in test_concatenation_as_lvalue() local
H A Dcxx11_tensor_device.cu294 Eigen::TensorMap<Eigen::Tensor<float, 3> > gpu_out(d_out, 40,50,70); in test_gpu() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/testing/
H A Ddelegate_testing.cc80 const float* gpu_out = gpu->typed_output_tensor<float>(i); in CompareCPUGPUResults() local
86 int8_t* gpu_out = gpu->typed_output_tensor<int8_t>(i); in CompareCPUGPUResults() local
92 uint8_t* gpu_out = gpu->typed_output_tensor<uint8_t>(i); in CompareCPUGPUResults() local
H A Dinternal_api_samples.cc60 const float* gpu_out = gpu[i].data(); in CompareCPUGPUResults() local
/aosp_15_r20/external/eigen/unsupported/doc/examples/SYCL/
H A DCwiseMul.cpp44 TensorMap<Tensor<DataType, 3, DataLayout, IndexType>> gpu_out(gpu_out_data, tensorRange); in main() local