Home
last modified time | relevance | path

Searched refs:hasAttrOfType (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dembed_tf_framework_pass.cc33 return !func->hasAttrOfType<UnitAttr>(TFFrameworkDialect::kTFEntryAttrName); in IsNotInsideTfEntryFunction()
61 if (!op->hasAttrOfType<UnitAttr>(TFFrameworkDialect::kTFEntryAttrName)) { in runOnOperation()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_extract_outside_compilation.cc64 if (walked_op->hasAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) { in HasOutsideCompilationNested()
75 if (op->hasAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) { in HasOutsideCompilationAncestor()
271 !op.hasAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) in MoveOpsToHost()
412 if (op->hasAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) { in RemoveOutsideCompilation()
Doutside_compiled_to_host_launch.cc93 if (op->hasAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) in runOnOperation()
Dmark_ops_for_outside_compilation.cc231 if (op->hasAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) in MarkVariantInputsOutputs()
244 !input_defining_op->hasAttrOfType<StringAttr>( in MarkVariantInputsOutputs()
Dregion_control_flow_to_functional.cc313 if (if_region->hasAttrOfType<StringAttr>(kThenFuncNameAttr) && in ConvertIfOp()
324 if (if_region->hasAttrOfType<StringAttr>(kElseFuncNameAttr) && in ConvertIfOp()
/external/llvm-project/mlir/include/mlir/IR/
DOperation.h338 bool hasAttrOfType(NameT &&name) { in hasAttrOfType() function