Searched refs:product_dims (Results 1 – 2 of 2) sorted by relevance
53 const Eigen::array<Eigen::IndexPair<int>, 1> product_dims = { in UpdateDenseDeltaWeights() local59 product_dims)) in UpdateDenseDeltaWeights()214 const Eigen::array<Eigen::IndexPair<int>, 1> product_dims = { in ComputeWxAndWeightedExampleNorm() local218 .contract(dense_vector.RowAsMatrix(), product_dims); in ComputeWxAndWeightedExampleNorm()221 .contract(dense_vector.RowAsMatrix(), product_dims); in ComputeWxAndWeightedExampleNorm()
1019 array<IndexPair<int>, 1> product_dims = { IndexPair(1, 0) };1020 Eigen::Tensor<int, 2> AB = a.contract(b, product_dims);