Searched defs:reduce_dim (Results 1 – 8 of 8) sorted by relevance
33 bool reduce_dim; in ApplyGradientTreesPredictionShapeFn() local49 bool reduce_dim; in ApplyGradientTreesPredictionVerboseShapeFn() local
169 reduce_dim=False): argument190 reduce_dim=False): argument
48 def _joined_array(num_dims, reduce_dim): argument
68 for (int32 reduce_dim : reduced_indices) { in GetReductionIterSize() local
425 Eigen::DSizes<Eigen::DenseIndex, REDUCENDIM> reduce_dim; in HandleReduce() local
118 bool reduce_dim; in GradientTreesPredictionOp() local
1329 const int64 reduce_dim = rhs_kept_dim < rhs_collapsing_dim in HandleDotStrengthReduction() local
1047 int reduce_dim = in TEST_F() local1063 int reduce_dim = in TEST_F() local