Searched refs:product_dims (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/tools/ |
| D | aot_compiled_test.cc | 52 const Eigen::array<Eigen::IndexPair<int>, 1> product_dims = { in TEST() local
56 arg_feed_x.contract(arg_feed_y, product_dims); in TEST()
86 const Eigen::array<Eigen::IndexPair<int>, 1> product_dims = { in TEST() local
90 arg_feed_x.contract(arg_feed_y, product_dims); in TEST()
115 const Eigen::array<Eigen::IndexPair<int>, 1> product_dims = { in TEST() local
119 arg_feed_x.contract(arg_feed_y, product_dims); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sdca_internal.cc | 53 const Eigen::array<Eigen::IndexPair<int>, 1> product_dims = { in UpdateDenseDeltaWeights() local
59 product_dims)) in UpdateDenseDeltaWeights()
214 const Eigen::array<Eigen::IndexPair<int>, 1> product_dims = { in ComputeWxAndWeightedExampleNorm() local
218 .contract(dense_vector.RowAsMatrix(), product_dims); in ComputeWxAndWeightedExampleNorm()
221 .contract(dense_vector.RowAsMatrix(), product_dims); in ComputeWxAndWeightedExampleNorm()
|
| /external/eigen/unsupported/Eigen/CXX11/src/Tensor/ |
| D | README.md | 1019 array<IndexPair<int>, 1> product_dims = { IndexPair(1, 0) };
1020 Eigen::Tensor<int, 2> AB = a.contract(b, product_dims);
|