Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/
Dconv_emitter_transforms.h29 mlir::AffineMap affine_map; member
Dconv_emitter_transforms.cc120 BoundAffineMap affine_map = GetBoundAffineMapFrom(owner); in TileLoop() local
Dconv_emitter.cc63 mlir::AffineMap affine_map; member
180 BoundAffineMap affine_map = GetBoundAffineMapFrom(owner); in HoistAndFix() local
/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/
Dtf_jitrt_tile_reduction.cc168 for (auto affine_map : makeArrayRef(indexing_maps).drop_back()) { in matchAndRewrite() local