Searched defs:reduce_dim (Results 1 – 7 of 7) sorted by relevance
76 Eigen::array<int, 1> reduce_dim({0}); in Compute() local127 Eigen::array<int, 1> reduce_dim({0}); in Compute() local
68 for (int32 reduce_dim : reduced_indices) { in GetReductionIterSize() local
492 Eigen::DSizes<Eigen::DenseIndex, REDUCENDIM> reduce_dim; in HandleReduce() local
48 def _joined_array(num_dims, reduce_dim): argument
92 int64 reduce_dim = n_dims - 1; in GetMatrixDiagonal() local
1048 int reduce_dim = in TEST_F() local1064 int reduce_dim = in TEST_F() local
2715 auto reduce_dim = rewriter.create<TF::ConstOp>( in matchAndRewrite() local