Searched refs:depth_dim (Results 1 – 2 of 2) sorted by relevance
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();