Home
last modified time | relevance | path

Searched refs:init_element_shape_ptrs (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dshape_inference.cc3768 absl::InlinedVector<const Shape*, 1> init_element_shape_ptrs; in InferScatterShape() local
3771 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()