Searched refs:init_value_shapes (Results 1 – 4 of 4) sorted by relevance
63 absl::Span<const Shape* const> init_value_shapes, in VerifyReducerShape() argument123 *init_value_shapes[i])) { in VerifyReducerShape()128 ShapeUtil::HumanString(*init_value_shapes[i])); in VerifyReducerShape()
1328 absl::InlinedVector<const Shape*, 2> init_value_shapes() const { in init_value_shapes() function
897 auto init_shapes = reduce_window_instr->init_value_shapes(); in HandleReduceWindow()
1953 reduce_window_instr->init_value_shapes(), window,