Searched refs:tensor5d (Results 1 – 2 of 2) sorted by relevance
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_morphing_sycl.cpp | 97 Tensor<DataType, 5, DataLayout, IndexType> tensor5d(dim3); in test_reshape_as_lvalue() local
103 …DataType* gpu_data3 = static_cast<DataType*>(sycl_device.allocate(tensor5d.size()*sizeof(DataType… in test_reshape_as_lvalue()
115 sycl_device.memcpyDeviceToHost(tensor5d.data(), gpu_data3,(tensor5d.size())*sizeof(DataType)); in test_reshape_as_lvalue()
121 VERIFY_IS_EQUAL(tensor5d(i,j,0,k,0), tensor(i,j,k)); in test_reshape_as_lvalue()
|
| D | cxx11_tensor_morphing.cpp | 102 TensorMap<Tensor<float, 5>> tensor5d(scratch, 2,3,1,7,1); in test_reshape_as_lvalue() local
103 tensor5d.reshape(dim).device(Eigen::DefaultDevice()) = tensor; in test_reshape_as_lvalue()
109 VERIFY_IS_EQUAL(tensor5d(i,j,0,k,0), tensor(i,j,k)); in test_reshape_as_lvalue()
|