fa08cfd489
This contains all performance hints to the API. Currently there's only bytes_per_pack, which splits large batches into multiple packs allows overlapping communication and computation. Currently we can only pack if all Tensors in the batch have known shapes. PiperOrigin-RevId: 297269428 Change-Id: Iaf7d7d3adf7c6cad59aa6079fbcd36b31e92c4b5
28 lines
530 B
Plaintext
28 lines
530 B
Plaintext
path: "tensorflow.distribute.experimental"
|
|
tf_module {
|
|
member {
|
|
name: "CentralStorageStrategy"
|
|
mtype: "<type \'type\'>"
|
|
}
|
|
member {
|
|
name: "CollectiveCommunication"
|
|
mtype: "<class \'enum.EnumMeta\'>"
|
|
}
|
|
member {
|
|
name: "CollectiveHints"
|
|
mtype: "<type \'type\'>"
|
|
}
|
|
member {
|
|
name: "MultiWorkerMirroredStrategy"
|
|
mtype: "<type \'type\'>"
|
|
}
|
|
member {
|
|
name: "ParameterServerStrategy"
|
|
mtype: "<type \'type\'>"
|
|
}
|
|
member {
|
|
name: "TPUStrategy"
|
|
mtype: "<type \'type\'>"
|
|
}
|
|
}
|