Home
last modified time | relevance | path

Searched refs:swap_layout (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Davgpooling_op.h35 To32Bit(output).swap_layout().device(d) = Eigen::SpatialAvgPooling( in operator()
36 To32Bit(input).swap_layout(), window_cols, window_rows, col_stride, in operator()
40 output.swap_layout().device(d) = Eigen::SpatialAvgPooling( in operator()
41 input.swap_layout(), window_cols, window_rows, col_stride, row_stride, in operator()
Dmaxpooling_op.h36 output.swap_layout().device(d) = in operator()
37 Eigen::SpatialMaxPooling(input.swap_layout(), window_cols, window_rows, in operator()
Dattention_ops.cc121 output->tensor<float, 4>().swap_layout().device( in Compute()
123 Eigen::ExtractGlimpses(input.tensor<float, 4>().swap_layout(), in Compute()
Dsparse_tensor_dense_matmul_op.cc306 b.swap_layout().shuffle(shuffle).conjugate(); in Compute()
/external/eigen/unsupported/test/
Dcxx11_tensor_layout_swap.cpp21 Tensor<float, 3, RowMajor> tensor2 = tensor.swap_layout(); in test_simple_swap()
42 tensor2.swap_layout() = tensor; in test_swap_as_lvalue()
Dcxx11_tensor_image_patch.cpp20 Tensor<float, 4, RowMajor> tensor_row_major = tensor.swap_layout(); in test_simple_patch()
206 Tensor<float, 4, RowMajor> tensor_row_major = tensor.swap_layout(); in test_patch_padding_valid()
279 Tensor<float, 4, RowMajor> tensor_row_major = tensor.swap_layout(); in test_patch_padding_valid_same_value()
355 Tensor<float, 4, RowMajor> tensor_row_major = tensor.swap_layout(); in test_patch_padding_same()
412 Tensor<float, 3, RowMajor> tensor_row_major = tensor.swap_layout(); in test_patch_no_extra_dim()
571 Tensor<float, 5, RowMajor> l_out_row_major = l_in.swap_layout().extract_image_patches(11, 11); in test_imagenet_patches()
621 l_out_row_major = l_in.swap_layout().extract_image_patches(9, 9); in test_imagenet_patches()
667 l_out_row_major = l_in.swap_layout().extract_image_patches(7, 7); in test_imagenet_patches()
713 l_out_row_major = l_in.swap_layout().extract_image_patches(3, 3); in test_imagenet_patches()
Dcxx11_tensor_volume_patch.cpp11 Tensor<float, 5, RowMajor> tensor_row_major = tensor.swap_layout(); in test_single_voxel_patch()
49 Tensor<float, 5, RowMajor> tensor_row_major = tensor.swap_layout(); in test_entire_volume_patch()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorBase.h733 swap_layout() const { in swap_layout() function
885 swap_layout() const { in swap_layout() function
890 swap_layout() { in swap_layout() function
DREADME.md139 ```swap_layout()``` method. Note that this will also reverse the order of the
149 col_major_result = row_major.swap_layout();
155 col_major_result = row_major.swap_layout().shuffle(shuffle);
1680 Tensor<float, 4, RowMajor> tensor_row_major = tensor.swap_layout();