Searched refs:init_element_shape_ptrs (Results 1 – 1 of 1) sorted by relevance
3768 absl::InlinedVector<const Shape*, 1> init_element_shape_ptrs; in InferScatterShape() local3771 init_element_shape_ptrs.reserve(operand_count); in InferScatterShape()3776 init_element_shape_ptrs.push_back(&init_element_shapes.back()); in InferScatterShape()3779 TF_RETURN_IF_ERROR(VerifyReducerShape(to_apply_shape, init_element_shape_ptrs, in InferScatterShape()