Searched refs:kernelChannels (Results 1 – 5 of 5) sorted by relevance
122 const TensorIndex kernelChannels = variable168 pre_contract_dims[0] = kernelChannels * kernelRows * kernelCols;174 pre_contract_dims[1] = kernelChannels * kernelRows * kernelCols;207 kernel_dims[1] = kernelChannels * kernelRows * kernelCols;209 kernel_dims[0] = kernelChannels * kernelRows * kernelCols;
129 const TensorIndex kernelChannels = variable216 kernel_dims[1] = kernelChannels;219 kernel_dims[0] = kernelChannels;257 post_contract_dims[0] = kernelChannels;264 post_contract_dims[NumDims - 1] = kernelChannels;449 const TensorIndex kernelChannels = variable517 input_dims[0] = kernelChannels;520 input_dims[1] = kernelChannels;545 post_contract_dims[0] = kernelChannels;553 post_contract_dims[3] = kernelChannels;
119 const TensorIndex kernelChannels = variable213 kernel_dims[1] = kernelChannels;216 kernel_dims[0] = kernelChannels;257 post_contract_dims[0] = kernelChannels;265 post_contract_dims[NumDims - 1] = kernelChannels;442 const TensorIndex kernelChannels = variable512 input_dims[0] = kernelChannels;515 input_dims[1] = kernelChannels;542 post_contract_dims[0] = kernelChannels;552 post_contract_dims[4] = kernelChannels;
1864 const TensorIndex kernelChannels =1876 eigen_assert(kernelChannels == in.dimension(0));1878 eigen_assert(kernelChannels == in.dimension(NumDims - 1));1917 kernel_dims[1] = kernelChannels * kernelPlanes * kernelRows * kernelCols;1919 kernel_dims[0] = kernelChannels * kernelPlanes * kernelRows * kernelCols;1930 kernelChannels * kernelPlanes * kernelRows * kernelCols;1937 kernelChannels * kernelPlanes * kernelRows * kernelCols;
1629 const TensorIndex kernelChannels =1681 pre_contract_dims[0] = kernelChannels * kernelRows * kernelCols;1687 pre_contract_dims[1] = kernelChannels * kernelRows * kernelCols;1720 kernel_dims[1] = kernelChannels * kernelRows * kernelCols;1722 kernel_dims[0] = kernelChannels * kernelRows * kernelCols;