Searched defs:tensor_load (Results 1 – 2 of 2) sorted by relevance
48 auto tensor_load = extract.tensor().getDefiningOp<memref::TensorLoadOp>(); in matchAndRewrite() local66 auto tensor_load = shape_of.arg().getDefiningOp<memref::TensorLoadOp>(); in matchAndRewrite() local
99 if (auto tensor_load = dyn_cast<memref::TensorLoadOp>(definingOp)) { in runOnFunction() local