Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/tests/
Dconv_depthwise_common.cc42 "_kernel_dims_", absl::StrJoin(spec.kernel_dims, "x"), "_kernel_layout_", in DepthwiseConvolution2DTestDataToString()
61 if (spec.activation_dims[1] == 1 && spec.kernel_dims[1] == 2) { in BuildHloTextDepthwiseConvolution2D()
75 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextDepthwiseConvolution2D()
81 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextDepthwiseConvolution2D()
99 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextDepthwiseConvolution2D()
105 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextDepthwiseConvolution2D()
123 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextDepthwiseConvolution2D()
129 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextDepthwiseConvolution2D()
Dgrouped_convolution_test.cc44 std::vector<int64> kernel_dims; member
84 config.kernel_dims = {kernel_size, kernel_size, group_size, output_feature}; in GetConv2DTestCases()
133 "_kernel_dims_", absl::StrJoin(spec.kernel_dims, "x"), "_kernel_layout_", in GroupedConvolution2DTestDataToString()
150 if (spec.activation_dims[1] == 1 && spec.kernel_dims[1] == 2) { in BuildHloTextGroupedConvolution2D()
166 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextGroupedConvolution2D()
172 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextGroupedConvolution2D()
192 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextGroupedConvolution2D()
198 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextGroupedConvolution2D()
217 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextGroupedConvolution2D()
223 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextGroupedConvolution2D()
Dconv_depthwise_backprop_filter_test.cc37 std::vector<int64> kernel_dims; member
76 config.kernel_dims = {batch, kernel_size, kernel_size, in GetConv2DTestCases()
102 config.kernel_dims = {batch, kernel_size / 2, kernel_size / 2, feature}; in GetConv2DTestCases()
127 "_kernel_dims_", absl::StrJoin(spec.kernel_dims, "x"), in BatchGroupedConvolution2DTestDataToString()
157 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextBatchGroupedConvolution2D()
163 absl::StrJoin(spec.kernel_dims, ","), in BuildHloTextBatchGroupedConvolution2D()
Dconv_depthwise_common.h38 std::vector<int64> kernel_dims; member
Dconv_depthwise_test.cc58 config.kernel_dims = {kernel_size, kernel_size, 1, feature}; in GetConv2DTestCases()
/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.h74 Eigen::DSizes<int64, 2> kernel_dims; in EigenConvImpl() local
75 kernel_dims[0] = kernel_channels * kernel_cols * kernel_rows; in EigenConvImpl()
76 kernel_dims[1] = kernel_filters; in EigenConvImpl()
87 .contract(kernel.reshape(kernel_dims), contract_dims) in EigenConvImpl()
/external/tensorflow/tensorflow/python/keras/utils/
Dconv_utils.py271 kernel_dims = len(kernel_shape)
273 if kernel_dims != in_dims or stride_dims != in_dims:
276 (stride_dims, in_dims, kernel_dims))
353 kernel_dims = len(kernel_shape)
355 if kernel_dims != in_dims or stride_dims != in_dims:
358 (stride_dims, in_dims, kernel_dims))
/external/ruy/ruy/
Dblock_map.cc452 block_map->kernel_dims[Side::kLhs] = kernel_rows; in MakeBlockMap()
453 block_map->kernel_dims[Side::kRhs] = kernel_cols; in MakeBlockMap()
477 block_map.kernel_dims[side]; in GetBlockMatrixCoords()
480 (block < block_map.large_blocks[side] ? block_map.kernel_dims[side] : 0); in GetBlockMatrixCoords()
482 RUY_DCHECK_EQ(0, *start % block_map.kernel_dims[side]); in GetBlockMatrixCoords()
483 RUY_DCHECK_EQ(0, *end % block_map.kernel_dims[side]); in GetBlockMatrixCoords()
Dblock_map_test.cc162 block_map.kernel_dims[side] = 1 << kKernelSizeLog2; in GetBlockByIndexSquareTest()
163 block_map.small_block_dims[side] = block_map.kernel_dims[side]; in GetBlockByIndexSquareTest()
Dblock_map.h97 SidePair<int> kernel_dims; member
Dtrace.h672 block_map->kernel_dims[Side::kLhs], \
674 block_map->kernel_dims[Side::kRhs]); \
680 block_map->kernel_dims[Side::kLhs], \
687 block_map->kernel_dims[Side::kLhs], \
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc331 std::vector<int64> kernel_dims; member
652 d.kernel_dims.resize(num_spatial_dims); in ChooseWindowedSpatialDims()
662 d.kernel_dims[i] = random_int(generator()), in ChooseWindowedSpatialDims()
663 d.input_dims[i] = RandomDim(d.kernel_dims[i]); in ChooseWindowedSpatialDims()
665 std::uniform_int_distribution<int>(1, d.kernel_dims[i])(generator()); in ChooseWindowedSpatialDims()
667 s = GetWindowedOutputSize(d.input_dims[i], d.kernel_dims[i], in ChooseWindowedSpatialDims()
1136 std::vector<int64> input_dims, kernel_dims, stride_dims; in TEST_F() local
1138 kernel_dims.push_back( in TEST_F()
1152 .Attr("ksize", ImageDims(FORMAT_NHWC, 1, 1, kernel_dims)) in TEST_F()
1173 .Attr("ksize", ImageDims(FORMAT_NHWC, 1, 1, d.kernel_dims)) in TEST_F()
[all …]
/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_spatial_convolutions.h213 DSizes<TensorIndex, 2> kernel_dims; variable
215 kernel_dims[0] = kernelFilters * kernelRows * kernelCols;
216 kernel_dims[1] = kernelChannels;
218 kernel_dims[1] = kernelFilters * kernelRows * kernelCols;
219 kernel_dims[0] = kernelChannels;
282 .reshape(kernel_dims)
302 .reshape(kernel_dims),
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.h1717 DSizes<TensorIndex, 2> kernel_dims;
1719 kernel_dims[0] = kernelFilters;
1720 kernel_dims[1] = kernelChannels * kernelRows * kernelCols;
1722 kernel_dims[0] = kernelChannels * kernelRows * kernelCols;
1723 kernel_dims[1] = kernelFilters;
1728 kernel.reshape(kernel_dims)
1748 .contract(kernel.reshape(kernel_dims), contract_dims, output_kernel)
1753 kernel.reshape(kernel_dims)
1766 .contract(kernel.reshape(kernel_dims), contract_dims, output_kernel)
Deigen_cuboid_convolution.h1914 DSizes<TensorIndex, 2> kernel_dims;
1916 kernel_dims[0] = kernelFilters;
1917 kernel_dims[1] = kernelChannels * kernelPlanes * kernelRows * kernelCols;
1919 kernel_dims[0] = kernelChannels * kernelPlanes * kernelRows * kernelCols;
1920 kernel_dims[1] = kernelFilters;
1975 kernel.reshape(kernel_dims)
1988 .contract(kernel.reshape(kernel_dims), contract_dims)
/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 …]