| /external/tensorflow/tensorflow/python/autograph/lang/ |
| directives.py | 51 swap_memory=UNSPECIFIED,
62 swap_memory: See tf.while_loop.
67 del swap_memory
|
| /external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
| rnn.py | 48 swap_memory=False,
118 swap_memory: Transparently swap the tensors produced in forward inference
264 swap_memory=swap_memory,
287 swap_memory=False,
326 swap_memory: Transparently swap the tensors produced in forward inference
377 swap_memory=swap_memory,
416 swap_memory=swap_memory,
[all...] |
| /external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
| rnn.py | 135 swap_memory=False):
175 swap_memory: Transparently swap the tensors produced in forward inference
238 swap_memory=swap_memory,
|
| /external/tensorflow/tensorflow/contrib/seq2seq/python/ops/ |
| decoder.py | 160 swap_memory=False,
166 self.swap_memory = swap_memory
177 swap_memory=self.swap_memory,
273 swap_memory=False,
295 swap_memory: Argument passed to `tf.while_loop`.
455 swap_memory=swap_memory)
|
| beam_search_decoder.py | [all...] |
| /external/tensorflow/tensorflow/python/ops/ |
| functional_ops.py | 48 swap_memory=False, name=None):
78 swap_memory: (optional) True enables GPU-CPU memory swapping.
144 swap_memory=swap_memory,
157 swap_memory=False, name=None):
187 swap_memory: (optional) True enables GPU-CPU memory swapping.
255 swap_memory=swap_memory,
268 swap_memory=False, infer_shape=True, reverse=False, name=None):
317 swap_memory: (optional) True enables GPU-CPU memory swapping
[all...] |
| map_fn.py | 39 swap_memory=False, infer_shape=True, name=None):
114 swap_memory: (optional) True enables GPU-CPU memory swapping.
267 swap_memory=swap_memory,
|
| rnn.py | 358 swap_memory=False, time_major=False, scope=None):
395 swap_memory: Transparently swap the tensors produced in forward inference
442 parallel_iterations=parallel_iterations, swap_memory=swap_memory,
474 parallel_iterations=parallel_iterations, swap_memory=swap_memory,
492 dtype=None, parallel_iterations=None, swap_memory=False,
564 swap_memory: Transparently swap the tensors produced in forward inference
669 swap_memory=swap_memory,
[all...] |
| control_flow_ops.py | 725 swap_memory=forward_ctxt.swap_memory,
739 swap_memory=forward_ctxt.swap_memory,
2302 def swap_memory(self): member in class:WhileContext
[all...] |
| /external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/ |
| dynamic_rnn_estimator.py | 219 swap_memory=True):
233 swap_memory: Transparently swap the tensors produced in forward inference
250 swap_memory=swap_memory,
400 swap_memory=True,
444 swap_memory: Transparently swap the tensors produced in forward inference
499 swap_memory=swap_memory)
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| ragged_map_ops.py | 42 swap_memory=False,
120 swap_memory: (optional) True enables GPU-CPU memory swapping.
302 swap_memory=swap_memory,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| rnn_test.py | [all...] |
| stack_ops_test.py | 56 c = gen_data_flow_ops.stack_push_v2(h, x, swap_memory=True)
78 v = gen_data_flow_ops.stack_push_v2(h, a, swap_memory=True)
196 c = gen_data_flow_ops.stack_push(h, x, swap_memory=True)
230 v = gen_data_flow_ops.stack_push(h, a, swap_memory=True)
|
| control_flow_ops_py_test.py | [all...] |
| /external/tensorflow/tensorflow/contrib/gan/python/eval/python/ |
| classifier_metrics_impl.py | 354 swap_memory=True,
512 swap_memory=True,
[all...] |
| summaries_impl.py | 272 swap_memory=True),
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| stack_ops_test.py | 48 c = gen_data_flow_ops.stack_push_v2(h, x, swap_memory=True)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| one_device_strategy.py | 126 parallel_iterations=1, back_prop=False, swap_memory=False,
|
| mirrored_strategy.py | 622 parallel_iterations=1, back_prop=False, swap_memory=False,
[all...] |
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| pfor.py | [all...] |
