cwise_ops_common.h | 71 int ndims; member in struct:tensorflow::BinaryOpShared::BinaryOpState 102 const int ndims = state.ndims; variable 106 if (ndims <= 1) { 123 } else if (ndims == 2) { 130 } else if (ndims == 3) { 137 } else if (ndims == 4) { 144 } else if (ndims == 5) { 281 // Partial specialization of BinaryFunctor<Device=CPUDevice, Functor, NDIMS> 283 template <typename Functor, int NDIMS> [all...] |