Searched defs:kernel_dims (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_backward_spatial_convolutions.h | 212 DSizes<TensorIndex, 2> kernel_dims; variable
465 DSizes<TensorIndex, 4> kernel_dims; variable
|
| D | eigen_backward_cuboid_convolutions.h | 210 DSizes<TensorIndex, 2> kernel_dims; variable
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | eigen_spatial_convolutions.h | 204 DSizes<TensorIndex, 2> kernel_dims; variable
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | eigen_spatial_convolutions.h | 207 DSizes<TensorIndex, 2> kernel_dims; variable
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_conv2d_impl.h | 68 Eigen::DSizes<int64, 2> kernel_dims; in EigenConvImpl() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | conv_depthwise_test.cc | 39 std::vector<int64> kernel_dims; member
|
| D | conv_depthwise_backprop_filter_test.cc | 38 std::vector<int64> kernel_dims; member
|
| D | grouped_convolution_test.cc | 40 std::vector<int64> kernel_dims; member
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | TensorConvolution.h | 27 IndexMapper(const InputDims& input_dims, const array<Index, NumKernelDims>& kernel_dims, in IndexMapper()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 330 std::vector<int64> kernel_dims; member
1135 std::vector<int64> input_dims, kernel_dims, stride_dims; in TEST_F() local
1606 std::vector<int64> kernel_dims = {d.kernel_dims[0], d.kernel_dims[1], in TEST_F() local
1785 std::vector<int64> kernel_dims = {d.kernel_dims[0], d.kernel_dims[1], in TEST_F() local
2339 std::vector<int64> input_dims, kernel_dims, stride_dims; in TEST_F() local
|