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

  /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
eigen_spatial_convolutions.h 125 const TensorIndex kernelChannels =
171 pre_contract_dims[0] = kernelChannels * kernelRows * kernelCols;
177 pre_contract_dims[1] = kernelChannels * kernelRows * kernelCols;
210 kernel_dims[1] = kernelChannels * kernelRows * kernelCols;
212 kernel_dims[0] = kernelChannels * kernelRows * kernelCols;
  /external/tensorflow/tensorflow/core/kernels/
eigen_cuboid_convolution.h 97 const TensorIndex kernelChannels =
109 eigen_assert(kernelChannels == in.dimension(0));
111 eigen_assert(kernelChannels == in.dimension(NumDims - 1));
150 kernel_dims[1] = kernelChannels * kernelDepth * kernelRows * kernelCols;
152 kernel_dims[0] = kernelChannels * kernelDepth * kernelRows * kernelCols;
163 kernelChannels * kernelDepth * kernelRows * kernelCols;
170 kernelChannels * kernelDepth * kernelRows * kernelCols;
eigen_backward_spatial_convolutions.h 128 const TensorIndex kernelChannels =
210 kernel_dims[1] = kernelChannels;
213 kernel_dims[0] = kernelChannels;
251 post_contract_dims[0] = kernelChannels;
258 post_contract_dims[NumDims - 1] = kernelChannels;
399 const TensorIndex kernelChannels =
439 pre_contract_dims[0] = kernelChannels * kernelRows * kernelCols;
446 pre_contract_dims[1] = kernelCols * kernelRows * kernelChannels;
466 kernel_dims[1] = kernelChannels;
471 kernel_dims[2] = kernelChannels;
    [all...]
eigen_backward_cuboid_convolutions.h 112 const TensorIndex kernelChannels =
197 kernel_dims[1] = kernelChannels;
201 kernel_dims[1] = kernelChannels;
247 post_contract_dims[0] = kernelChannels;
255 post_contract_dims[NumDims - 1] = kernelChannels;
411 const TensorIndex kernelChannels =
492 input_dims[0] = kernelChannels;
500 input_dims[2] = kernelChannels;
533 kernel_dims[0] = kernelChannels;
543 kernel_dims[4] = kernelChannels;
    [all...]

Completed in 279 milliseconds