Home
last modified time | relevance | path

Searched refs:post_contract_dims (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_cuboid_convolutions.h255 DSizes<TensorIndex, NumDims> post_contract_dims; variable
257 post_contract_dims[0] = kernelChannels;
258 post_contract_dims[1] = inputPlanes;
259 post_contract_dims[2] = inputRows;
260 post_contract_dims[3] = inputCols;
262 post_contract_dims[i] = out.dimension(i);
265 post_contract_dims[NumDims - 1] = kernelChannels;
266 post_contract_dims[NumDims - 2] = inputPlanes;
267 post_contract_dims[NumDims - 3] = inputRows;
268 post_contract_dims[NumDims - 4] = inputCols;
[all …]
Deigen_backward_spatial_convolutions.h255 DSizes<TensorIndex, NumDims> post_contract_dims; variable
257 post_contract_dims[0] = kernelChannels;
258 post_contract_dims[1] = inputRows;
259 post_contract_dims[2] = inputCols;
261 post_contract_dims[i] = out.dimension(i);
264 post_contract_dims[NumDims - 1] = kernelChannels;
265 post_contract_dims[NumDims - 2] = inputRows;
266 post_contract_dims[NumDims - 3] = inputCols;
268 post_contract_dims[i] = out.dimension(i);
291 .reshape(post_contract_dims),
[all …]
Deigen_spatial_convolutions-inl.h1700 DSizes<TensorIndex, NumDims> post_contract_dims;
1702 post_contract_dims[0] = kernelFilters;
1703 post_contract_dims[1] = out_height;
1704 post_contract_dims[2] = out_width;
1706 post_contract_dims[i] = in.dimension(i);
1709 post_contract_dims[NumDims - 1] = kernelFilters;
1710 post_contract_dims[NumDims - 2] = out_height;
1711 post_contract_dims[NumDims - 3] = out_width;
1713 post_contract_dims[i] = in.dimension(i);
1739 .reshape(post_contract_dims),
[all …]
Deigen_cuboid_convolution.h1954 DSizes<TensorIndex, NumDims> post_contract_dims;
1956 post_contract_dims[0] = kernelFilters;
1957 post_contract_dims[1] = out_planes;
1958 post_contract_dims[2] = out_height;
1959 post_contract_dims[3] = out_width;
1961 post_contract_dims[i] = in.dimension(i);
1964 post_contract_dims[NumDims - 1] = kernelFilters;
1965 post_contract_dims[NumDims - 2] = out_planes;
1966 post_contract_dims[NumDims - 3] = out_height;
1967 post_contract_dims[NumDims - 4] = out_width;
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Deigen_spatial_convolutions.h187 DSizes<TensorIndex, NumDims> post_contract_dims; variable
189 post_contract_dims[0] = kernelFilters;
190 post_contract_dims[1] = out_height;
191 post_contract_dims[2] = out_width;
193 post_contract_dims[i] = in.dimension(i);
196 post_contract_dims[NumDims - 1] = kernelFilters;
197 post_contract_dims[NumDims - 2] = out_height;
198 post_contract_dims[NumDims - 3] = out_width;
200 post_contract_dims[i] = in.dimension(i);
219 .reshape(post_contract_dims);
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_conv2d_impl.h68 Eigen::DSizes<int64, 4> post_contract_dims; in EigenConvImpl() local
69 post_contract_dims[0] = input_batch; in EigenConvImpl()
70 post_contract_dims[1] = output_rows; in EigenConvImpl()
71 post_contract_dims[2] = output_cols; in EigenConvImpl()
72 post_contract_dims[3] = kernel_filters; in EigenConvImpl()
88 .reshape(post_contract_dims); in EigenConvImpl()