Searched defs:reduce_dims (Results 1 – 8 of 8) 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
99 Eigen::array<int, 1> reduce_dims({0}); in operator ()() local104 Eigen::IndexList<Eigen::type2index<0> > reduce_dims; in operator ()() local191 Eigen::array<int, 1> reduce_dims({0}); in operator ()() local196 Eigen::IndexList<Eigen::type2index<0> > reduce_dims; in operator ()() local
81 Eigen::array<int, 1> reduce_dims({0}); in Compute() local
466 std::vector<int64> reduce_dims; in XLA_TEST_P() local567 std::vector<int64> reduce_dims; in XLA_TEST_P() local678 std::vector<int64> reduce_dims; in XLA_TEST_P() local
549 std::vector<int64> reduce_dims; member
1752 std::vector<int64> reduce_dims( in HandleDot() local2994 std::vector<int64> reduce_dims = reduce->dimensions(); in HandleReduce() local3118 absl::InlinedVector<int64, 8> reduce_dims; in HandleReduceWindow() local
622 BCast::Vec reduce_dims[2]; in MaterializeBroadcastGradientArgs() local