Searched refs:two_dims (Results 1 – 1 of 1) sorted by relevance
295 Eigen::DSizes<Eigen::Index, 2> two_dims(batch * height * width * depth, in Compute() local297 redux(context->eigen_device<Device>(), two_dims, output_backprop, in Compute()