| /external/tensorflow/tensorflow/core/kernels/ |
| avgpooling_op.h | 35 To32Bit(output).swap_layout().device(d) = Eigen::SpatialAvgPooling(
36 To32Bit(input).swap_layout(), window_cols, window_rows, col_stride,
40 output.swap_layout().device(d) = Eigen::SpatialAvgPooling(
41 input.swap_layout(), window_cols, window_rows, col_stride, row_stride,
|
| maxpooling_op.h | 36 output.swap_layout().device(d) =
37 Eigen::SpatialMaxPooling(input.swap_layout(), window_cols, window_rows,
|
| attention_ops.cc | 97 output->tensor<float, 4>().swap_layout().device(
99 Eigen::ExtractGlimpses(input.tensor<float, 4>().swap_layout(),
|
| sparse_tensor_dense_matmul_op.cc | 306 b.swap_layout().shuffle(shuffle).conjugate();
|
| /external/eigen/unsupported/test/ |
| cxx11_tensor_layout_swap.cpp | 21 Tensor<float, 3, RowMajor> tensor2 = tensor.swap_layout();
42 tensor2.swap_layout() = tensor;
|
| cxx11_tensor_volume_patch.cpp | 11 Tensor<float, 5, RowMajor> tensor_row_major = tensor.swap_layout();
49 Tensor<float, 5, RowMajor> tensor_row_major = tensor.swap_layout();
|
| cxx11_tensor_image_patch.cpp | 20 Tensor<float, 4, RowMajor> tensor_row_major = tensor.swap_layout();
206 Tensor<float, 4, RowMajor> tensor_row_major = tensor.swap_layout();
279 Tensor<float, 4, RowMajor> tensor_row_major = tensor.swap_layout();
355 Tensor<float, 4, RowMajor> tensor_row_major = tensor.swap_layout();
412 Tensor<float, 3, RowMajor> tensor_row_major = tensor.swap_layout();
571 Tensor<float, 5, RowMajor> l_out_row_major = l_in.swap_layout().extract_image_patches(11, 11);
621 l_out_row_major = l_in.swap_layout().extract_image_patches(9, 9);
667 l_out_row_major = l_in.swap_layout().extract_image_patches(7, 7);
713 l_out_row_major = l_in.swap_layout().extract_image_patches(3, 3);
[all...] |
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| TensorBase.h | 733 swap_layout() const { function in class:Eigen::TensorBase
885 swap_layout() const { function in class:Eigen::TensorBase
890 swap_layout() { function in class:Eigen::TensorBase
[all...] |
