Searched defs:reduce_dims (Results 1 – 13 of 13) sorted by relevance
73 array<DenseIndex, 3> reduce_dims; in test_argmax_tuple_reducer() local105 array<DenseIndex, 3> reduce_dims; in test_argmin_tuple_reducer() local
57 absl::flat_hash_set<std::string> reduce_dims; in ExpandOp() local389 const Layout& output_layout, absl::flat_hash_set<std::string>& reduce_dims, in MaybeRelayoutInputs()
236 Eigen::IndexList<Eigen::type2index<0>, Eigen::type2index<2> > reduce_dims; in Compute() local
174 Eigen::IndexList<Eigen::type2index<0>> reduce_dims; in operator ()() local
529 std::vector<int64_t> reduce_dims; in XLA_TEST_P() local630 std::vector<int64_t> reduce_dims; in XLA_TEST_P() local741 std::vector<int64_t> reduce_dims; in XLA_TEST_P() local
557 std::vector<int64_t> reduce_dims; member
273 std::vector<int64_t> reduce_dims; in EinsumDiagonalLabels() local
2727 std::vector<int64_t> reduce_dims(dnums.lhs_contracting_dimensions_size()); in HandleDot() local5421 std::vector<int64_t> reduce_dims = *reduce->mutable_dimensions(); in HandleReduce() local5540 std::vector<int64_t> reduce_dims; in HandleReduce() local5746 DimensionVector reduce_dims; in HandleReduceWindow() local
1630 auto reduce_dims = consumer->dimensions(); in SupportedOpForPropagation() local
1051 tensorflow::BCast::Vec reduce_dims[2]; in matchAndRewrite() local
1885 SmallVector<int64_t, 4> reduce_dims; in matchAndRewrite() local
680 BCast::Vec reduce_dims[2]; in MaterializeBroadcastGradientArgs() local
472 DenseIntElementsAttr reduce_dims, in ApplyReduction()2263 auto reduce_dims = GetI64ElementsAttr(non_feature_dims, &rewriter); in matchAndRewrite() local