Add get_ prefix to parallelism_threads APIs
PiperOrigin-RevId: 251391606
This commit is contained in:
parent
0b1f96ed96
commit
0bdb97031c
@ -22,7 +22,7 @@ from tensorflow.python.eager import context
|
|||||||
from tensorflow.python.util.tf_export import tf_export
|
from tensorflow.python.util.tf_export import tf_export
|
||||||
|
|
||||||
|
|
||||||
@tf_export('config.threading.intra_op_parallelism_threads')
|
@tf_export('config.threading.get_intra_op_parallelism_threads')
|
||||||
def get_intra_op_parallelism_threads():
|
def get_intra_op_parallelism_threads():
|
||||||
"""Get number of threads used within an individual op for parallelism.
|
"""Get number of threads used within an individual op for parallelism.
|
||||||
|
|
||||||
@ -50,7 +50,7 @@ def set_intra_op_parallelism_threads(num_threads):
|
|||||||
context.context().intra_op_parallelism_threads = num_threads
|
context.context().intra_op_parallelism_threads = num_threads
|
||||||
|
|
||||||
|
|
||||||
@tf_export('config.threading.inter_op_parallelism_threads')
|
@tf_export('config.threading.get_inter_op_parallelism_threads')
|
||||||
def get_inter_op_parallelism_threads():
|
def get_inter_op_parallelism_threads():
|
||||||
"""Get number of threads used for parallelism between independent operations.
|
"""Get number of threads used for parallelism between independent operations.
|
||||||
|
|
||||||
|
@ -1,11 +1,11 @@
|
|||||||
path: "tensorflow.config.threading"
|
path: "tensorflow.config.threading"
|
||||||
tf_module {
|
tf_module {
|
||||||
member_method {
|
member_method {
|
||||||
name: "inter_op_parallelism_threads"
|
name: "get_inter_op_parallelism_threads"
|
||||||
argspec: "args=[], varargs=None, keywords=None, defaults=None"
|
argspec: "args=[], varargs=None, keywords=None, defaults=None"
|
||||||
}
|
}
|
||||||
member_method {
|
member_method {
|
||||||
name: "intra_op_parallelism_threads"
|
name: "get_intra_op_parallelism_threads"
|
||||||
argspec: "args=[], varargs=None, keywords=None, defaults=None"
|
argspec: "args=[], varargs=None, keywords=None, defaults=None"
|
||||||
}
|
}
|
||||||
member_method {
|
member_method {
|
||||||
|
@ -1,11 +1,11 @@
|
|||||||
path: "tensorflow.config.threading"
|
path: "tensorflow.config.threading"
|
||||||
tf_module {
|
tf_module {
|
||||||
member_method {
|
member_method {
|
||||||
name: "inter_op_parallelism_threads"
|
name: "get_inter_op_parallelism_threads"
|
||||||
argspec: "args=[], varargs=None, keywords=None, defaults=None"
|
argspec: "args=[], varargs=None, keywords=None, defaults=None"
|
||||||
}
|
}
|
||||||
member_method {
|
member_method {
|
||||||
name: "intra_op_parallelism_threads"
|
name: "get_intra_op_parallelism_threads"
|
||||||
argspec: "args=[], varargs=None, keywords=None, defaults=None"
|
argspec: "args=[], varargs=None, keywords=None, defaults=None"
|
||||||
}
|
}
|
||||||
member_method {
|
member_method {
|
||||||
|
Loading…
x
Reference in New Issue
Block a user