Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/passes/
Dtensorexpr_fuser.cpp60 Value* broadcastSizes(at::ArrayRef<Value*> sizes, AliasDb* db) { in broadcastSizes() function
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc2544 auto broadcastSizes = rewriter.getI64TensorAttr(resultTy.getShape()); in matchAndRewrite() local