Home
last modified time | relevance | path

Searched defs:dim_expr (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/
H A Dtf_jitrt_buffer_forwarding.cc130 auto dim_expr = expr.dyn_cast<mlir::AffineDimExpr>(); in matchAndRewrite() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
H A Dbuffer_reuse_pass.cc208 auto dim_expr = expr.dyn_cast<AffineDimExpr>(); in can_reuse_locally() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/tensorexpr/
H A Deval.cpp847 for (const ExprPtr& dim_expr : b->dims()) { in visit() local
917 for (const auto& dim_expr : b->dims()) { in visit() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/
H A Dtf_cpurt_passes.cc181 auto dim_expr = expr.dyn_cast<mlir::AffineDimExpr>(); in matchAndRewrite() local