Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtpu_rewrite_device_util.cc498 auto num_cores_per_replica_attr = cluster->getAttrOfType<mlir::IntegerAttr>( in GetHostDeviceOutsideComputation() local
500 if (!num_cores_per_replica_attr) in GetHostDeviceOutsideComputation()
504 if (num_cores_per_replica_attr.getInt() != 1) in GetHostDeviceOutsideComputation()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_cluster_formation.cc569 auto num_cores_per_replica_attr = in FormClustersInBlock() local
573 if (num_cores_per_replica_attr) in FormClustersInBlock()
574 num_cores_per_replica = num_cores_per_replica_attr.getInt(); in FormClustersInBlock()
Dtpu_rewrite_pass.cc576 auto num_cores_per_replica_attr = cluster_func->getAttrOfType<IntegerAttr>( in Rewrite() local
578 if (!num_cores_per_replica_attr) in Rewrite()
582 int num_cores_per_replica = num_cores_per_replica_attr.getInt(); in Rewrite()