Searched defs:reduce_dims (Results 1 – 11 of 11) 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
106 std::vector<int64> reduce_dims(out_backprop_shape.dims() - 1); in Compile() local
165 Eigen::array<int, 1> reduce_dims({0}); in operator ()() local170 Eigen::IndexList<Eigen::type2index<0>> reduce_dims; in operator ()() local302 Eigen::array<int, 1> reduce_dims({0}); in operator ()() local
564 std::vector<int64> reduce_dims; in XLA_TEST_P() local665 std::vector<int64> reduce_dims; in XLA_TEST_P() local776 std::vector<int64> reduce_dims; in XLA_TEST_P() local
550 std::vector<int64> reduce_dims; member
2436 std::vector<int64> reduce_dims( in HandleDot() local4621 std::vector<int64> reduce_dims = reduce->dimensions(); in HandleReduce() local4740 std::vector<int64> reduce_dims; in HandleReduce() local4802 absl::InlinedVector<int64, 8> reduce_dims; in HandleReduceWindow() local
1133 auto reduce_dims = consumer->dimensions(); in SupportedOpForPropagation() local
242 std::vector<int64> reduce_dims; in EinsumDiagonalLabels() local
627 SmallVector<int64_t, 4> reduce_dims; in matchAndRewrite() local
596 DenseIntElementsAttr reduce_dims, in ApplyReduction()1861 auto reduce_dims = GetI64ElementsAttr(non_feature_dims, &rewriter); in matchAndRewrite() local
676 BCast::Vec reduce_dims[2]; in MaterializeBroadcastGradientArgs() local