Searched defs:scatter_indices_shape (Results 1 – 3 of 3) sorted by relevance
33 const Shape& scatter_indices_shape = scatter_indices->shape(); in TransposeIndexVectorDimToLast() local115 const Shape& scatter_indices_shape, HloInstruction* updates, in AdjustScatterDims()332 const Shape& scatter_indices_shape = scatter_indices->shape(); in ScatterTripCount() local
3582 const Shape& operand_shape, absl::Span<const int64> scatter_indices_shape, in ValidateScatterDimensionNumbers()3676 const Shape& operand_shape, const Shape& scatter_indices_shape, in InferScatterShape()
410 Shape scatter_indices_shape; member