Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dmark_ops_for_outside_compilation.cc43 constexpr char kXlaOutsideCompilationAttr[] = "_xla_outside_compilation"; variable
264 if (parent->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) in HasOutsideCompiledAncestor()
277 if (op->hasAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) in MarkVariantInputsOutputs()
291 kXlaOutsideCompilationAttr)) { in MarkVariantInputsOutputs()
293 kXlaOutsideCompilationAttr, in MarkVariantInputsOutputs()
305 !user->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) { in MarkVariantInputsOutputs()
306 user->setAttr(kXlaOutsideCompilationAttr, in MarkVariantInputsOutputs()
334 op->setAttr(kXlaOutsideCompilationAttr, in MarkUncompilableOps()
357 if (iter_op->hasAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) { in ContainsUncompilableOps()
377 if (!op->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) return; in UnmarkChildren()
[all …]
Doutside_compiled_to_host_launch.cc31 constexpr char kXlaOutsideCompilationAttr[] = "_xla_outside_compilation"; variable
54 host_op->removeAttr(StringAttr::get(context, kXlaOutsideCompilationAttr)); in WrapOpInLaunch()
65 if (op->hasAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) { in runOnOperation()
Dtpu_host_computation_expansion.cc36 constexpr char kXlaOutsideCompilationAttr[] = "_xla_outside_compilation"; variable
39 return op->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr) != nullptr; in HasOutsideCompilationAttribute()
108 head_outside_compiled_op->setAttr(kXlaOutsideCompilationAttr, in ExpandHeadOutsideCompiledOps()
Dtpu_update_embedding_enqueue_op_inputs.cc39 constexpr char kXlaOutsideCompilationAttr[] = "_xla_outside_compilation"; variable
132 embedding_op->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr); in UpdateEmbeddingEnqueueOpInput()
134 enqueue_mode->setAttr(kXlaOutsideCompilationAttr, in UpdateEmbeddingEnqueueOpInput()
Dhost_launch_to_outside_compiled.cc35 constexpr char kXlaOutsideCompilationAttr[] = "_xla_outside_compilation"; variable
46 op->setAttr(kXlaOutsideCompilationAttr, in MarkOutsideCompiledInRegion()
Dtpu_extract_outside_compilation.cc56 constexpr char kXlaOutsideCompilationAttr[] = "_xla_outside_compilation"; variable
122 if (walked_op->hasAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) { in HasOutsideCompilationNested()
133 if (op->hasAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) { in HasOutsideCompilationAncestor()
415 op->setAttr(kXlaOutsideCompilationAttr, in MarkOutsideCompiled()
602 !op.hasAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) in MoveOpsToHost()
711 if (op->hasAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) { in RemoveOutsideCompilation()
713 StringAttr::get(op->getContext(), kXlaOutsideCompilationAttr)); in RemoveOutsideCompilation()
Dtpu_extract_head_tail_outside_compilation.cc55 constexpr char kXlaOutsideCompilationAttr[] = "_xla_outside_compilation"; variable
58 return op->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr) != nullptr; in HasOutsideCompilationAttribute()
191 head_outside_compiled_op->removeAttr(kXlaOutsideCompilationAttr); in CreateHeadComputation()
303 tail_outside_compiled_op->removeAttr(kXlaOutsideCompilationAttr); in CreateTailComputation()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_defs.h37 inline constexpr absl::string_view kXlaOutsideCompilationAttr = variable
54 kXlaOutsideCompilationAttr,
Dfunctionalize_cond.cc960 if (GetNodeAttr(if_node->def(), kXlaOutsideCompilationAttr, in AddIdentityNode()
963 id_builder.Attr(kXlaOutsideCompilationAttr, outside_compilation); in AddIdentityNode()
/external/tensorflow/tensorflow/compiler/jit/
Dcompilability_check_util.cc80 return n.attrs().Find(kXlaOutsideCompilationAttr) != nullptr; in IsInOutsideCompilationCluster()