Home
last modified time | relevance | path

Searched refs:HasValidCompilationAndReplicationAttributes (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dattribute_utils.cc30 LogicalResult HasValidCompilationAndReplicationAttributes(Operation& op) { in HasValidCompilationAndReplicationAttributes() function
H A Dattribute_utils.h91 LogicalResult HasValidCompilationAndReplicationAttributes(Operation &op);
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dconvert_to_legacy_compile_and_replicate_attributes.cc45 if (failed(TF::HasValidCompilationAndReplicationAttributes(*op))) in ConvertToLegacyAttributes()
H A Dtpu_cluster_formation.cc142 LogicalResult result = TF::HasValidCompilationAndReplicationAttributes(op); in CollectAndGroupClusterOps()
H A Dtpu_rewrite_pass.cc902 if (failed(TF::HasValidCompilationAndReplicationAttributes(*op))) in runOnOperation()