Home
last modified time | relevance | path

Searched refs:post_contract_dims (Results 1 – 7 of 7) 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.h254 DSizes<TensorIndex, NumDims> post_contract_dims; variable
256 post_contract_dims[0] = kernelChannels;
257 post_contract_dims[1] = inputRows;
258 post_contract_dims[2] = inputCols;
260 post_contract_dims[i] = out.dimension(i);
263 post_contract_dims[NumDims - 1] = kernelChannels;
264 post_contract_dims[NumDims - 2] = inputRows;
265 post_contract_dims[NumDims - 3] = inputCols;
267 post_contract_dims[i] = out.dimension(i);
285 .reshape(post_contract_dims),
[all …]
Deigen_cuboid_convolution.h1760 DSizes<TensorIndex, NumDims> post_contract_dims;
1762 post_contract_dims[0] = kernelFilters;
1763 post_contract_dims[1] = out_planes;
1764 post_contract_dims[2] = out_height;
1765 post_contract_dims[3] = out_width;
1767 post_contract_dims[i] = in.dimension(i);
1770 post_contract_dims[NumDims - 1] = kernelFilters;
1771 post_contract_dims[NumDims - 2] = out_planes;
1772 post_contract_dims[NumDims - 3] = out_height;
1773 post_contract_dims[NumDims - 4] = out_width;
[all …]
Deigen_spatial_convolutions-inl.h1451 DSizes<TensorIndex, NumDims> post_contract_dims;
1453 post_contract_dims[0] = kernelFilters;
1454 post_contract_dims[1] = out_height;
1455 post_contract_dims[2] = out_width;
1457 post_contract_dims[i] = in.dimension(i);
1460 post_contract_dims[NumDims - 1] = kernelFilters;
1461 post_contract_dims[NumDims - 2] = out_height;
1462 post_contract_dims[NumDims - 3] = out_width;
1464 post_contract_dims[i] = in.dimension(i);
1485 .reshape(post_contract_dims),
[all …]
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Deigen_spatial_convolutions.h190 DSizes<TensorIndex, NumDims> post_contract_dims; variable
192 post_contract_dims[0] = kernelFilters;
193 post_contract_dims[1] = out_height;
194 post_contract_dims[2] = out_width;
196 post_contract_dims[i] = in.dimension(i);
199 post_contract_dims[NumDims - 1] = kernelFilters;
200 post_contract_dims[NumDims - 2] = out_height;
201 post_contract_dims[NumDims - 3] = out_width;
203 post_contract_dims[i] = in.dimension(i);
222 .reshape(post_contract_dims);
/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.h62 Eigen::DSizes<int64, 4> post_contract_dims; in EigenConvImpl() local
63 post_contract_dims[0] = input_batch; in EigenConvImpl()
64 post_contract_dims[1] = output_rows; in EigenConvImpl()
65 post_contract_dims[2] = output_cols; in EigenConvImpl()
66 post_contract_dims[3] = kernel_filters; in EigenConvImpl()
82 .reshape(post_contract_dims); in EigenConvImpl()