Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc2318 matching_non_contracting_dims = in LhsIsBestMatchForNonContractingPartitioning() local
2366 output_dims.reserve(matching_non_contracting_dims.size()); in LhsIsBestMatchForNonContractingPartitioning()
2368 matching_non_contracting_dims) { in LhsIsBestMatchForNonContractingPartitioning()
2376 matching_non_contracting_dims); in LhsIsBestMatchForNonContractingPartitioning()
2383 output_sharding, matching_non_contracting_dims, in LhsIsBestMatchForNonContractingPartitioning()