bd20260350
- This adds a field in virtual device options to specify priority for virtual devices. - When the priority is specified, it will be used to create the cuda streams for the virtual device with the given priority. - This is backwards compatible with no priorities specified. When no priorities specified, the current implementation of creating a stream without any priority will continue while any non-zero priorities specified will be used to create streams with that priority. PiperOrigin-RevId: 314470276 Change-Id: I4943f71e901245fb21b6f7e833adbdcd8126f1fa
24 lines
567 B
Plaintext
24 lines
567 B
Plaintext
path: "tensorflow.config.experimental.VirtualDeviceConfiguration"
|
|
tf_class {
|
|
is_instance: "<class \'tensorflow.python.eager.context.LogicalDeviceConfiguration\'>"
|
|
is_instance: "<class \'tensorflow.python.eager.context.LogicalDeviceConfiguration\'>"
|
|
is_instance: "<type \'tuple\'>"
|
|
member {
|
|
name: "experimental_priority"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member {
|
|
name: "memory_limit"
|
|
mtype: "<type \'property\'>"
|
|
}
|
|
member_method {
|
|
name: "__init__"
|
|
}
|
|
member_method {
|
|
name: "count"
|
|
}
|
|
member_method {
|
|
name: "index"
|
|
}
|
|
}
|