/external/tensorflow/tensorflow/core/kernels/ |
D | eigen_backward_spatial_convolutions.h | 212 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 …]
|
D | eigen_backward_cuboid_convolutions.h | 210 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)
|
D | eigen_spatial_convolutions-inl.h | 1468 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)
|
D | eigen_cuboid_convolution.h | 1720 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/ |
D | conv_depthwise_test.cc | 39 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()
|
D | grouped_convolution_test.cc | 40 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()
|
D | conv_depthwise_backprop_filter_test.cc | 38 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/ |
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/lite/kernels/internal/optimized/ |
D | eigen_spatial_convolutions.h | 204 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/ |
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 | 330 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/ |
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 …]
|
/external/tensorflow/tensorflow/python/keras/utils/ |
D | conv_utils.py | 284 kernel_dims = len(kernel_shape) 286 if kernel_dims != in_dims or stride_dims != in_dims: 289 (stride_dims, in_dims, kernel_dims))
|