Searched refs:tensor_map (Results 1 – 3 of 3) sorted by relevance
/external/eigen/unsupported/test/ |
D | cxx11_tensor_scan.cpp | 89 TensorMap<Tensor<int, 1, DataLayout> > tensor_map(inputs, 20); in test_tensor_maps() local 90 tensor_map.setRandom(); in test_tensor_maps() 92 Tensor<int, 1, DataLayout> result = tensor_map.cumsum(0); in test_tensor_maps() 96 accum += tensor_map(i); in test_tensor_maps()
|
D | cxx11_tensor_io.cpp | 112 TensorMap<Tensor<const int, 1, DataLayout> > tensor_map(tensor.data(), 5); in test_output_const() local 115 os << tensor_map; in test_output_const()
|
D | cxx11_tensor_reduction.cpp | 334 TensorMap<Tensor<int, 4, DataLayout> > tensor_map(inputs, 2, 3, 5, 7); in test_tensor_maps() local 340 tensor_map.setRandom(); in test_tensor_maps() 345 Tensor<int, 2, DataLayout> result = tensor_map.sum(reduction_axis); in test_tensor_maps() 355 sum += tensor_map(i, k, j, l); in test_tensor_maps()
|