Home
last modified time | relevance | path

Searched defs:num_cores_per_replica_attr (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtpu_rewrite_device_util.cc486 mlir::IntegerAttr num_cores_per_replica_attr = in HasModelParallelism() local
507 auto num_cores_per_replica_attr = cluster->getAttrOfType<mlir::IntegerAttr>( in GetHostDeviceOutsideComputation() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_cluster_formation.cc632 auto num_cores_per_replica_attr = in FormClustersInBlock() local
Dtpu_rewrite_pass.cc724 auto num_cores_per_replica_attr = cluster_func->getAttrOfType<IntegerAttr>( in Rewrite() local