cwise_ops_common.h | 69 int ndims; member in struct:tensorflow::BinaryOpShared::BinaryOpState 100 const int ndims = state.ndims; variable 104 if (ndims <= 1) { 121 } else if (ndims == 2) { 128 } else if (ndims == 3) { 135 } else if (ndims == 4) { 142 } else if (ndims == 5) { 278 // Partial specialization of BinaryFunctor<Device=CPUDevice, Functor, NDIMS> 280 template <typename Functor, int NDIMS> [all...] |