Searched refs:kernel_dims (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | eigen_backward_cuboid_convolutions.h | 194 DSizes<TensorIndex, 3> kernel_dims; variable 196 kernel_dims[0] = kernelFilters; 197 kernel_dims[1] = kernelChannels; 198 kernel_dims[2] = kernelRows * kernelCols * kernelPlanes; 200 kernel_dims[0] = kernelRows * kernelCols * kernelPlanes; 201 kernel_dims[1] = kernelChannels; 202 kernel_dims[2] = kernelFilters; 281 .reshape(kernel_dims) 297 .contract(kernel.reverse(kernel_reverse).reshape(kernel_dims), 531 DSizes<TensorIndex, 5> kernel_dims; variable [all …]
|
D | eigen_backward_spatial_convolutions.h | 207 DSizes<TensorIndex, 2> kernel_dims; variable 209 kernel_dims[0] = kernelFilters * kernelRows * kernelCols; 210 kernel_dims[1] = kernelChannels; 212 kernel_dims[1] = kernelFilters * kernelRows * kernelCols; 213 kernel_dims[0] = kernelChannels; 270 .reshape(kernel_dims) 289 .reshape(kernel_dims) 463 DSizes<TensorIndex, 4> kernel_dims; variable 465 kernel_dims[0] = kernelFilters; 466 kernel_dims[1] = kernelChannels; [all …]
|
D | eigen_cuboid_convolution.h | 147 DSizes<TensorIndex, 2> kernel_dims; variable 149 kernel_dims[0] = kernelFilters; 150 kernel_dims[1] = kernelChannels * kernelDepth * kernelRows * kernelCols; 152 kernel_dims[0] = kernelChannels * kernelDepth * kernelRows * kernelCols; 153 kernel_dims[1] = kernelFilters; 208 kernel.reshape(kernel_dims) 221 .contract(kernel.reshape(kernel_dims), contract_dims)
|
D | eigen_spatial_convolutions.h | 1042 DSizes<TensorIndex, 2> kernel_dims; 1044 kernel_dims[0] = kernelFilters; 1045 kernel_dims[1] = kernelChannels * kernelRows * kernelCols; 1047 kernel_dims[0] = kernelChannels * kernelRows * kernelCols; 1048 kernel_dims[1] = kernelFilters; 1054 kernel.reshape(kernel_dims) 1066 .contract(kernel.reshape(kernel_dims), contract_dims)
|
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
D | eigen_spatial_convolutions.h | 207 DSizes<TensorIndex, 2> kernel_dims; variable 209 kernel_dims[0] = kernelFilters; 210 kernel_dims[1] = kernelChannels * kernelRows * kernelCols; 212 kernel_dims[0] = kernelChannels * kernelRows * kernelCols; 213 kernel_dims[1] = kernelFilters; 221 .contract(kernel.reshape(kernel_dims), contract_dims)
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
D | runtime_conv2d_impl.h | 68 Eigen::DSizes<int64, 2> kernel_dims; in EigenConvImpl() local 69 kernel_dims[0] = kernel_channels * kernel_cols * kernel_rows; in EigenConvImpl() 70 kernel_dims[1] = kernel_filters; in EigenConvImpl() 81 .contract(kernel.reshape(kernel_dims), contract_dims) in EigenConvImpl()
|
/external/tensorflow/tensorflow/compiler/tests/ |
D | randomized_tests.cc | 305 std::vector<int64> kernel_dims; member 580 d.kernel_dims.resize(num_spatial_dims); in ChooseWindowedSpatialDims() 590 d.kernel_dims[i] = random_int(generator()), in ChooseWindowedSpatialDims() 591 d.input_dims[i] = RandomDim(d.kernel_dims[i]); in ChooseWindowedSpatialDims() 593 std::uniform_int_distribution<int>(1, d.kernel_dims[i])(generator()); in ChooseWindowedSpatialDims() 595 s = GetWindowedOutputSize(d.input_dims[i], d.kernel_dims[i], in ChooseWindowedSpatialDims() 1047 std::vector<int64> input_dims, kernel_dims, stride_dims; in TEST_F() local 1049 kernel_dims.push_back( in TEST_F() 1063 .Attr("ksize", ImageDims(FORMAT_NHWC, 1, 1, kernel_dims)) in TEST_F() 1084 .Attr("ksize", ImageDims(FORMAT_NHWC, 1, 1, d.kernel_dims)) in TEST_F() [all …]
|
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
D | TensorConvolution.h | 27 IndexMapper(const InputDims& input_dims, const array<Index, NumKernelDims>& kernel_dims, in IndexMapper() argument 34 const Index kernel_dim = kernel_dims[i]; in IndexMapper() 319 …const typename TensorEvaluator<KernelArgType, Device>::Dimensions& kernel_dims = m_kernelImpl.dime… 338 const Index kernel_dim = kernel_dims[i]; 342 m_kernelStride[i] = m_kernelStride[i - 1] * kernel_dims[i - 1]; 357 const Index kernel_dim = kernel_dims[i]; 361 m_kernelStride[i] = m_kernelStride[i + 1] * kernel_dims[i + 1]; 778 …const typename TensorEvaluator<KernelArgType, GpuDevice>::Dimensions& kernel_dims = m_kernelImpl.d… 784 const Index kernel_dim = kernel_dims[i]; 904 const array<Index, 1> kernel_dims(m_kernelImpl.dimensions()[0]); [all …]
|