Home
last modified time | relevance | path

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

/external/pytorch/torch/csrc/jit/tensorexpr/operators/
Dreduction.cpp55 std::vector<ExprHandle> reductionDims; in computeSum() local
/external/tensorflow/tensorflow/compiler/xla/mlir_hlo/lib/Dialect/mhlo/transforms/
Dlegalize_to_linalg.cc114 ArrayRef<int64_t> reductionDims) { in getTransposeMapForReduction()
292 const llvm::SmallSetVector<StringRef, 4>& reductionDims) { in getEinsumLoopsAttrs()
1562 ArrayRef<int64_t> reductionDims) { in getReduceOpInitTensorDynSizes()
1617 SmallVector<int64_t, 4> reductionDims = extract1DVector(op.dimensions()); in matchAndRewrite() local