Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_spatial_convolutions.h212 DSizes<TensorIndex, 2> kernel_dims; variable
214 kernel_dims[0] = kernelFilters * kernelRows * kernelCols;
215 kernel_dims[1] = kernelChannels;
217 kernel_dims[1] = kernelFilters * kernelRows * kernelCols;
218 kernel_dims[0] = kernelChannels;
275 .reshape(kernel_dims)
294 .reshape(kernel_dims)
465 DSizes<TensorIndex, 4> kernel_dims; variable
467 kernel_dims[0] = kernelFilters;
468 kernel_dims[1] = kernelChannels;
[all …]
Deigen_backward_cuboid_convolutions.h210 DSizes<TensorIndex, 2> kernel_dims; variable
212 kernel_dims[0] = kernelFilters * kernelPlanes * kernelRows * kernelCols;
213 kernel_dims[1] = kernelChannels;
215 kernel_dims[1] = kernelFilters * kernelPlanes * kernelRows * kernelCols;
216 kernel_dims[0] = kernelChannels;
278 .reshape(kernel_dims)
297 .reshape(kernel_dims)
Deigen_spatial_convolutions-inl.h1468 DSizes<TensorIndex, 2> kernel_dims;
1470 kernel_dims[0] = kernelFilters;
1471 kernel_dims[1] = kernelChannels * kernelRows * kernelCols;
1473 kernel_dims[0] = kernelChannels * kernelRows * kernelCols;
1474 kernel_dims[1] = kernelFilters;
1478 kernel.reshape(kernel_dims)
1490 .contract(kernel.reshape(kernel_dims), contract_dims, output_kernel)
Deigen_cuboid_convolution.h1720 DSizes<TensorIndex, 2> kernel_dims;
1722 kernel_dims[0] = kernelFilters;
1723 kernel_dims[1] = kernelChannels * kernelPlanes * kernelRows * kernelCols;
1725 kernel_dims[0] = kernelChannels * kernelPlanes * kernelRows * kernelCols;
1726 kernel_dims[1] = kernelFilters;
1781 kernel.reshape(kernel_dims)
1794 .contract(kernel.reshape(kernel_dims), contract_dims)
/external/tensorflow/tensorflow/compiler/xla/tests/
Dconv_depthwise_test.cc39 std::vector<int64> kernel_dims; member
71 config.kernel_dims = {kernel_size, kernel_size, 1, feature}; in GetConv2DTestCases()
119 "_kernel_dims_", absl::StrJoin(spec.kernel_dims, "x"), "_kernel_layout_", in DepthwiseConvolution2DTestDataToString()
136 if (spec.activation_dims[1] == 1 && spec.kernel_dims[1] == 2) { in BuildHloTextDepthwiseConvolution2D()
151 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextDepthwiseConvolution2D()
157 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextDepthwiseConvolution2D()
176 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextDepthwiseConvolution2D()
182 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextDepthwiseConvolution2D()
200 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextDepthwiseConvolution2D()
206 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextDepthwiseConvolution2D()
Dgrouped_convolution_test.cc40 std::vector<int64> kernel_dims; member
80 config.kernel_dims = {kernel_size, kernel_size, group_size, output_feature}; in GetConv2DTestCases()
128 "_kernel_dims_", absl::StrJoin(spec.kernel_dims, "x"), "_kernel_layout_", in GroupedConvolution2DTestDataToString()
145 if (spec.activation_dims[1] == 1 && spec.kernel_dims[1] == 2) { in BuildHloTextGroupedConvolution2D()
161 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextGroupedConvolution2D()
167 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextGroupedConvolution2D()
187 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextGroupedConvolution2D()
193 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextGroupedConvolution2D()
212 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextGroupedConvolution2D()
218 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextGroupedConvolution2D()
Dconv_depthwise_backprop_filter_test.cc38 std::vector<int64> kernel_dims; member
69 config.kernel_dims = {batch, kernel_size, kernel_size, feature}; in GetConv2DTestCases()
90 config.kernel_dims = {batch, kernel_size / 2, kernel_size / 2, feature}; in GetConv2DTestCases()
115 "_kernel_dims_", absl::StrJoin(spec.kernel_dims, "x"), in BatchGroupedConvolution2DTestDataToString()
143 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextBatchGroupedConvolution2D()
149 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextBatchGroupedConvolution2D()
/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/lite/kernels/internal/optimized/
Deigen_spatial_convolutions.h204 DSizes<TensorIndex, 2> kernel_dims; variable
206 kernel_dims[0] = kernelFilters;
207 kernel_dims[1] = kernelChannels * kernelRows * kernelCols;
209 kernel_dims[0] = kernelChannels * kernelRows * kernelCols;
210 kernel_dims[1] = kernelFilters;
218 .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.cc330 std::vector<int64> kernel_dims; member
651 d.kernel_dims.resize(num_spatial_dims); in ChooseWindowedSpatialDims()
661 d.kernel_dims[i] = random_int(generator()), in ChooseWindowedSpatialDims()
662 d.input_dims[i] = RandomDim(d.kernel_dims[i]); in ChooseWindowedSpatialDims()
664 std::uniform_int_distribution<int>(1, d.kernel_dims[i])(generator()); in ChooseWindowedSpatialDims()
666 s = GetWindowedOutputSize(d.input_dims[i], d.kernel_dims[i], in ChooseWindowedSpatialDims()
1135 std::vector<int64> input_dims, kernel_dims, stride_dims; in TEST_F() local
1137 kernel_dims.push_back( in TEST_F()
1151 .Attr("ksize", ImageDims(FORMAT_NHWC, 1, 1, kernel_dims)) in TEST_F()
1172 .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 …]
/external/tensorflow/tensorflow/python/keras/utils/
Dconv_utils.py284 kernel_dims = len(kernel_shape)
286 if kernel_dims != in_dims or stride_dims != in_dims:
289 (stride_dims, in_dims, kernel_dims))