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()361 const Shape& scatter_indices_shape = scatter_indices->shape(); in ExpandScatter() local
3070 const Shape& operand_shape, absl::Span<const int64> scatter_indices_shape, in ValidateScatterDimensionNumbers()3164 const Shape& operand_shape, const Shape& scatter_indices_shape, in InferScatterShape()
1123 Shape scatter_indices_shape = scatter_indices->shape(); in EmitScatter() local