Searched defs:index_vector_dim (Results 1 – 7 of 7) sorted by relevance
32 HloInstruction* scatter_indices, int64 index_vector_dim) { in TransposeIndexVectorDimToLast()57 HloInstruction* scatter_indices, int64 index_vector_dim) { in CanonicalizeScatterIndices()116 int64 index_vector_dim) { in AdjustScatterDims()
30 HloInstruction* start_indices, int64 index_vector_dim) { in TransposeIndexVectorDimToLast()57 HloInstruction* start_indices, int64 index_vector_dim) { in CanonicalizeGatherIndices()88 int64 index_vector_dim) { in AdjustBatchDimsInAccumulator()
905 int64 index_vector_dim = dim_numbers_.index_vector_dim(); in FetchIndexVector() local1033 int64 index_vector_dim, const Literal& start_indices, in ReshapedGatherIndices()
2332 string index_vector_dim = StrCat( in GatherDimensionNumbersToString() local2343 absl::Span<const int64> start_index_map, int64 index_vector_dim) { in MakeGatherDimNumbers()2419 string index_vector_dim = StrCat( in ScatterDimensionNumbersToString() local2433 int64 index_vector_dim) { in MakeScatterDimNumbers()
2042 int64 index_vector_dim, const Literal& indices, in ReshapedScatterIndices()2179 int64 index_vector_dim = dim_numbers_.index_vector_dim(); in FetchIndexVector() local
1624 optional<int64> index_vector_dim; in ParseInstructionRhs() local1659 optional<int64> index_vector_dim; in ParseInstructionRhs() local
462 int64 index_vector_dim = 4; field476 int64 index_vector_dim = 4; field