Home
last modified time | relevance | path

Searched refs:kernel_dims (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_cuboid_convolutions.h194 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 …]
Deigen_backward_spatial_convolutions.h207 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 …]
Deigen_cuboid_convolution.h147 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)
Deigen_spatial_convolutions.h1042 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/
Deigen_spatial_convolutions.h207 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/
Druntime_conv2d_impl.h68 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/
Drandomized_tests.cc305 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/
DTensorConvolution.h27 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 …]