Home
last modified time | relevance | path

Searched defs:matmuls (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dbatch_mm.cpp219 std::vector<Node*> matmuls; in removeTransposesAndGatherMatmuls() local
305 auto matmuls = root.removeTransposesAndGatherMatmuls(); in BatchMMTreeReduce() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqmatmul.cpp106 auto matmuls = [&](int64_t begin, int64_t end) { in qmatmul() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dunroll_batch_matmul.cc247 std::vector<Value> matmuls; in matchAndRewrite() local