Searched defs:contracting_dims (Results 1 – 7 of 7) sorted by relevance
44 std::vector<DimNums> contracting_dims; member
976 int64 rank, absl::Span<const int64> contracting_dims, in GetOnlyNonContractingNonBatchDim()1001 absl::Span<const int64> contracting_dims, in CanFoldDotIntoIndexedArray()
603 absl::Span<const int64> batch_dims) -> bool { in ValidateDotDimensionNumbers()628 absl::Span<const int64> batch_dims) -> bool { in ValidateDotDimensionNumbers()
346 std::vector<int64> contracting_dims; in InferDotShardingFromOperands() local
394 std::vector<DimsMapping> contracting_dims; member
1752 -> absl::optional<PartitionedHlo> { in PartitionDotGroupOnBatch()
331 C* batch_dims, C* contracting_dims) { in DeleteDimsFromContainer()