Home
last modified time | relevance | path

Searched refs:is_fusion (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dcontraction_fusion.cc105 auto is_fusion = [&](const NamedAttribute &attr) -> bool { in matchAndRewrite() local
115 llvm::find_if(attrs, is_fusion)->second.template cast<ArrayAttr>(); in matchAndRewrite()
116 llvm::erase_if(attrs, is_fusion); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.h729 mlir::Region* region, bool is_fusion);
Dir_emitter_unnested.cc2736 bool is_fusion) { in GetOrCreateSubComputationFromRegion() argument
2740 if (is_fusion) { in GetOrCreateSubComputationFromRegion()
2753 if (is_fusion) { in GetOrCreateSubComputationFromRegion()
2762 if (is_fusion) { in GetOrCreateSubComputationFromRegion()