Home
last modified time | relevance | path

Searched refs:kXlaOutsideCompilationAttr (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dmark_ops_for_outside_compilation.cc43 constexpr char kXlaOutsideCompilationAttr[] = "_xla_outside_compilation"; variable
218 if (parent->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) in HasOutsideCompiledAncestor()
231 if (op->hasAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) in MarkVariantInputsOutputs()
245 kXlaOutsideCompilationAttr)) { in MarkVariantInputsOutputs()
247 kXlaOutsideCompilationAttr, in MarkVariantInputsOutputs()
259 !user->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) { in MarkVariantInputsOutputs()
260 user->setAttr(kXlaOutsideCompilationAttr, in MarkVariantInputsOutputs()
288 op->setAttr(kXlaOutsideCompilationAttr, in MarkUncompilableOps()
307 if (!op->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) return; in UnmarkChildren()
311 if (parent_op->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) { in UnmarkChildren()
[all …]
Doutside_compiled_to_host_launch.cc31 constexpr char kXlaOutsideCompilationAttr[] = "_xla_outside_compilation"; variable
76 host_op->removeAttr(Identifier::get(kXlaOutsideCompilationAttr, context)); in WrapOpInLaunch()
93 if (op->hasAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) in runOnOperation()
Dtpu_host_computation_expansion.cc38 constexpr char kXlaOutsideCompilationAttr[] = "_xla_outside_compilation"; variable
41 return op->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr) != nullptr; in HasOutsideCompilationAttribute()
110 head_outside_compiled_op->setAttr(kXlaOutsideCompilationAttr, in ExpandHeadOutsideCompiledOps()
Dtpu_update_embedding_enqueue_op_inputs.cc38 constexpr char kXlaOutsideCompilationAttr[] = "_xla_outside_compilation"; variable
129 embedding_op->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr); in UpdateEmbeddingEnqueueOpInput()
131 enqueue_mode->setAttr(kXlaOutsideCompilationAttr, in UpdateEmbeddingEnqueueOpInput()
Dtpu_extract_outside_compilation.cc51 constexpr char kXlaOutsideCompilationAttr[] = "_xla_outside_compilation"; variable
64 if (walked_op->hasAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) { in HasOutsideCompilationNested()
75 if (op->hasAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) { in HasOutsideCompilationAncestor()
254 op->setAttr(kXlaOutsideCompilationAttr, in MarkOutsideCompiled()
271 !op.hasAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) in MoveOpsToHost()
412 if (op->hasAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) { in RemoveOutsideCompilation()
414 Identifier::get(kXlaOutsideCompilationAttr, op->getContext())); in RemoveOutsideCompilation()
Dtpu_extract_head_tail_outside_compilation.cc53 constexpr char kXlaOutsideCompilationAttr[] = "_xla_outside_compilation"; variable
56 return op->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr) != nullptr; in HasOutsideCompilationAttribute()
186 head_outside_compiled_op->removeAttr(kXlaOutsideCompilationAttr); in CreateHeadComputation()
294 tail_outside_compiled_op->removeAttr(kXlaOutsideCompilationAttr); in CreateTailComputation()
/external/tensorflow/tensorflow/compiler/jit/
Dmark_for_compilation_pass.h35 extern const char* const kXlaOutsideCompilationAttr;
Dxla_cluster_util.h39 extern const char* const kXlaOutsideCompilationAttr;
Dxla_cluster_util.cc40 const char* const kXlaOutsideCompilationAttr = "_XlaOutsideCompilation"; variable