Searched refs:depth_dim (Results 1 – 5 of 5) sorted by relevance
49 Eigen::array<typename internal::traits<Input>::Index, 1> depth_dim; in eval() local50 depth_dim[0] = 0; in eval()60 Eigen::IndexList<Eigen::type2index<0> > depth_dim; in eval()71 input.maximum(depth_dim).eval().reshape(dims2d).broadcast(bcast)) * in eval()76 (output.sum(depth_dim).eval().reshape(dims2d).broadcast(bcast)); in eval()
38 Eigen::array<int, 1> depth_dim; in TEST() local39 depth_dim[0] = 0; in TEST()48 input.maximum(depth_dim).eval().reshape(dims2d).broadcast(bcast)) * in TEST()53 (reference.sum(depth_dim).eval().reshape(dims2d).broadcast(bcast)); in TEST()
56 Eigen::array<int, 1> depth_dim; in test_const() local57 depth_dim[0] = 0; in test_const()65 …Tensor<float, 2> output_tensor= (input_tensor - input_tensor.maximum(depth_dim).eval().reshape(dim… in test_const()
321 Tensor<...> Y = ((X - X.maximum(depth_dim).reshape(dims2d).broadcast(bcast))329 ((X - X.maximum(depth_dim).eval().reshape(dims2d).broadcast(bcast))338 Y = Y / (Y.sum(depth_dim).reshape(dims2d).broadcast(bcast));344 Y = Y / (Y.sum(depth_dim).eval().reshape(dims2d).broadcast(bcast));355 (Y / (Y.sum(depth_dim).eval().reshape(dims2d).broadcast(bcast))).eval();
765 DimensionHandle depth_dim = c->Dim( in AvgPoolShape() local783 {output_rows, output_cols}, depth_dim, in AvgPoolShape()