HomeSort by relevance Sort by last modified time
    Searched defs:post_contract_dims (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
runtime_conv2d_impl.h 62 Eigen::DSizes<int64, 4> post_contract_dims; local
63 post_contract_dims[0] = input_batch;
64 post_contract_dims[1] = output_rows;
65 post_contract_dims[2] = output_cols;
66 post_contract_dims[3] = kernel_filters;
82 .reshape(post_contract_dims);
  /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
eigen_spatial_convolutions.h 190 DSizes<TensorIndex, NumDims> post_contract_dims; local
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);
    [all...]
  /external/tensorflow/tensorflow/core/kernels/
eigen_cuboid_convolution.h 187 DSizes<TensorIndex, NumDims> post_contract_dims; local
189 post_contract_dims[0] = kernelFilters;
190 post_contract_dims[1] = out_depth;
191 post_contract_dims[2] = out_height;
192 post_contract_dims[3] = out_width;
194 post_contract_dims[i] = in.dimension(i);
197 post_contract_dims[NumDims - 1] = kernelFilters;
198 post_contract_dims[NumDims - 2] = out_depth;
199 post_contract_dims[NumDims - 3] = out_height;
200 post_contract_dims[NumDims - 4] = out_width
    [all...]
eigen_backward_spatial_convolutions.h 249 DSizes<TensorIndex, NumDims> post_contract_dims; local
251 post_contract_dims[0] = kernelChannels;
252 post_contract_dims[1] = inputRows;
253 post_contract_dims[2] = inputCols;
255 post_contract_dims[i] = out.dimension(i);
258 post_contract_dims[NumDims - 1] = kernelChannels;
259 post_contract_dims[NumDims - 2] = inputRows;
260 post_contract_dims[NumDims - 3] = inputCols;
262 post_contract_dims[i] = out.dimension(i);
280 .reshape(post_contract_dims),
    [all...]
eigen_backward_cuboid_convolutions.h 245 DSizes<TensorIndex, NumDims> post_contract_dims; local
247 post_contract_dims[0] = kernelChannels;
248 post_contract_dims[1] = inputPlanes;
249 post_contract_dims[2] = inputRows;
250 post_contract_dims[3] = inputCols;
252 post_contract_dims[i] = out.dimension(i);
255 post_contract_dims[NumDims - 1] = kernelChannels;
256 post_contract_dims[NumDims - 2] = inputPlanes;
257 post_contract_dims[NumDims - 3] = inputRows;
258 post_contract_dims[NumDims - 4] = inputCols
    [all...]

Completed in 260 milliseconds