Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Deigen_spatial_convolutions.h166 DSizes<TensorIndex, 2> pre_contract_dims; variable
168 pre_contract_dims[0] = kernelChannels * kernelRows * kernelCols;
169 pre_contract_dims[1] = out_height * out_width;
171 pre_contract_dims[1] *= in.dimension(i);
174 pre_contract_dims[1] = kernelChannels * kernelRows * kernelCols;
175 pre_contract_dims[0] = out_height * out_width;
177 pre_contract_dims[0] *= in.dimension(i);
217 .reshape(pre_contract_dims)
/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_spatial_convolutions.h227 DSizes<TensorIndex, 2> pre_contract_dims; variable
229 pre_contract_dims[0] = kernelFilters * kernelRows * kernelCols;
230 pre_contract_dims[1] = inputRows * inputCols;
232 pre_contract_dims[1] *= out.dimension(i);
235 pre_contract_dims[1] = kernelFilters * kernelRows * kernelCols;
236 pre_contract_dims[0] = inputRows * inputCols;
238 pre_contract_dims[0] *= out.dimension(i);
289 .reshape(pre_contract_dims),
297 .reshape(pre_contract_dims)
528 DSizes<TensorIndex, 2> pre_contract_dims; variable
[all …]
Deigen_backward_cuboid_convolutions.h225 DSizes<TensorIndex, 2> pre_contract_dims; variable
227 pre_contract_dims[0] =
229 pre_contract_dims[1] = inputPlanes * inputRows * inputCols;
231 pre_contract_dims[1] *= out.dimension(i);
234 pre_contract_dims[1] =
236 pre_contract_dims[0] = inputPlanes * inputRows * inputCols;
238 pre_contract_dims[0] *= out.dimension(i);
286 .reshape(pre_contract_dims),
294 .reshape(pre_contract_dims)
523 DSizes<TensorIndex, 2> pre_contract_dims; variable
[all …]
Deigen_spatial_convolutions-inl.h1679 DSizes<TensorIndex, 2> pre_contract_dims;
1681 pre_contract_dims[0] = kernelChannels * kernelRows * kernelCols;
1682 pre_contract_dims[1] = out_height * out_width;
1684 pre_contract_dims[1] *= in.dimension(i);
1687 pre_contract_dims[1] = kernelChannels * kernelRows * kernelCols;
1688 pre_contract_dims[0] = out_height * out_width;
1690 pre_contract_dims[0] *= in.dimension(i);
1737 .reshape(pre_contract_dims),
1747 .reshape(pre_contract_dims)
1758 .reshape(pre_contract_dims),
[all …]
Deigen_cuboid_convolution.h1927 DSizes<TensorIndex, 2> pre_contract_dims;
1929 pre_contract_dims[0] =
1931 pre_contract_dims[1] = out_planes * out_height * out_width;
1933 pre_contract_dims[1] *= in.dimension(i);
1936 pre_contract_dims[1] =
1938 pre_contract_dims[0] = out_planes * out_height * out_width;
1940 pre_contract_dims[0] *= in.dimension(i);
1980 .reshape(pre_contract_dims),
1987 .reshape(pre_contract_dims)
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_conv2d_impl.h59 Eigen::DSizes<int64, 2> pre_contract_dims; in EigenConvImpl() local
60 pre_contract_dims[0] = output_cols * output_rows * input_batch; in EigenConvImpl()
61 pre_contract_dims[1] = kernel_channels * kernel_cols * kernel_rows; in EigenConvImpl()
82 .reshape(pre_contract_dims) in EigenConvImpl()