Searched refs:for_op (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
| D | canonicalize.cc | 51 LogicalResult matchAndRewrite(scf::ForOp for_op, in matchAndRewrite()
53 Location loc = for_op.getLoc(); in matchAndRewrite()
55 if (!matchPattern(for_op.lowerBound(), m_ConstantInt(&lower_bound)) || in matchAndRewrite()
56 !matchPattern(for_op.upperBound(), m_ConstantInt(&upper_bound)) || in matchAndRewrite()
57 !matchPattern(for_op.step(), m_ConstantInt(&step))) { in matchAndRewrite()
65 Block *single_block = for_op.getBody(); in matchAndRewrite()
67 Value iv = for_op.getInductionVar(); in matchAndRewrite()
69 llvm::zip(for_op.getRegionIterArgs(), for_op.initArgs())) { in matchAndRewrite()
72 mapping.map(iv, for_op.lowerBound()); in matchAndRewrite()
90 for (auto iter_op : llvm::zip(for_op.getRegionIterArgs(), in matchAndRewrite()
[all …]
|
| /external/llvm-project/mlir/utils/spirv/ |
| D | gen_spirv_dialect.py | 229 def get_availability_spec(enum_case, capability_mapping, for_op, for_cap): argument
246 assert not (for_op and for_cap), 'cannot set both for_op and for_cap'
260 if for_op and not min_version:
268 if for_op and not max_version:
280 min_version = DEFAULT_MIN_VERSION if for_op else ''
283 if for_op and not exts:
310 if for_op and not caps:
318 for_op and min_version == DEFAULT_MIN_VERSION and
324 'let' if for_op else 'list<Availability>', joined_spec)
|