Home
last modified time | relevance | path

Searched defs:targetRank (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
Drank_specialization.cc328 int64_t targetRank = 0; in materializeRankedOperations() local
562 const SmallVector<Value, 8> &shapes, int64_t targetRank) { in materializeTargetRankSpecializationCase()
Dlegalize_to_linalg.cc228 auto targetRank = targetTy.getRank(); in matchAndRewrite() local
3172 auto targetRank = outputType.getRank(); in matchAndRewrite() local