Home
last modified time | relevance | path

Searched refs:swap_memory (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
Dfunctional_ops.py53 swap_memory=False, name=None): argument
130 swap_memory=swap_memory)
141 swap_memory=False, name=None): argument
218 swap_memory=swap_memory)
229 swap_memory=False, infer_shape=True, name=None): argument
413 swap_memory=swap_memory)
433 swap_memory=False, infer_shape=True, name=None): argument
618 back_prop=back_prop, swap_memory=swap_memory)
Drnn.py331 swap_memory=False, time_major=False, scope=None): argument
418 parallel_iterations=parallel_iterations, swap_memory=swap_memory,
444 parallel_iterations=parallel_iterations, swap_memory=swap_memory,
459 dtype=None, parallel_iterations=None, swap_memory=False, argument
633 swap_memory=swap_memory,
651 swap_memory, argument
832 swap_memory=swap_memory)
855 parallel_iterations=None, swap_memory=False, scope=None): argument
1157 swap_memory=swap_memory)
Dcontrol_flow_ops.py858 swap_memory=forward_ctxt.swap_memory,
872 swap_memory=forward_ctxt.swap_memory,
1027 swap_enabled = self.forward_context.swap_memory
1034 enter_acc, value, swap_memory=swap_enabled)
1049 enter_acc, value, swap_memory=swap_enabled)
1057 enter_acc, value, swap_memory=swap_enabled)
2199 swap_memory=False, argument
2223 swap_memory, name)
2228 swap_memory, name): argument
2248 self._swap_memory = swap_memory
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Drnn_test.py497 def _dynamic_rnn_swap_memory_benchmark(inputs_t, sequence_length, swap_memory): argument
509 swap_memory=swap_memory,
537 inputs_t, sequence_length, swap_memory=False)
545 inputs_t, sequence_length, swap_memory=True)
555 swap_memory, nn): argument
573 inputs_t, sequence_length, swap_memory=swap_memory)
709 swap_memory = True
711 if swap_memory:
713 dynamic, swap_memory, 2)
717 swap_memory, 3)
Dstack_ops_test.py54 c = gen_data_flow_ops._stack_push_v2(h, x, swap_memory=True)
75 v = gen_data_flow_ops._stack_push_v2(h, a, swap_memory=True)
187 c = gen_data_flow_ops._stack_push(h, x, swap_memory=True)
220 v = gen_data_flow_ops._stack_push(h, a, swap_memory=True)
Dcontrol_flow_ops_py_test.py3155 swap_memory=True)
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Ddynamic_rnn_estimator.py214 swap_memory=True): argument
245 swap_memory=swap_memory,
395 swap_memory=True, argument
494 swap_memory=swap_memory)
/external/tensorflow/tensorflow/contrib/seq2seq/python/ops/
Ddecoder.py155 swap_memory=False, argument
309 swap_memory=swap_memory)
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_StackPushV2.pbtxt23 name: "swap_memory"
/external/tensorflow/tensorflow/contrib/gan/python/eval/python/
Dclassifier_metrics_impl.py343 swap_memory=True,
501 swap_memory=True,
/external/tensorflow/tensorflow/core/protobuf/
Dcontrol_flow.proto50 bool swap_memory = 4; field
/external/tensorflow/tensorflow/contrib/bayesflow/python/ops/
Dmetropolis_hastings_impl.py315 parallel_iterations=1, swap_memory=1))
/external/tensorflow/tensorflow/tools/api/golden/
Dtensorflow.nn.pbtxt45 …tate_fw\', \'initial_state_bw\', \'dtype\', \'parallel_iterations\', \'swap_memory\', \'time_major…
129 …nce_length\', \'initial_state\', \'dtype\', \'parallel_iterations\', \'swap_memory\', \'time_major…
249 …argspec: "args=[\'cell\', \'loop_fn\', \'parallel_iterations\', \'swap_memory\', \'scope\'], varar…
Dtensorflow.pbtxt1065 … \'elems\', \'initializer\', \'parallel_iterations\', \'back_prop\', \'swap_memory\', \'name\'], v…
1069 … \'elems\', \'initializer\', \'parallel_iterations\', \'back_prop\', \'swap_memory\', \'name\'], v…
1337 …'fn\', \'elems\', \'dtype\', \'parallel_iterations\', \'back_prop\', \'swap_memory\', \'infer_shap…
1657 … \'elems\', \'initializer\', \'parallel_iterations\', \'back_prop\', \'swap_memory\', \'infer_shap…
2101 …ars\', \'shape_invariants\', \'parallel_iterations\', \'back_prop\', \'swap_memory\', \'name\', \'…
/external/tensorflow/tensorflow/compiler/tests/
Dstack_ops_test.py48 c = gen_data_flow_ops._stack_push_v2(h, x, swap_memory=True)
/external/tensorflow/tensorflow/python/keras/_impl/keras/
Dbackend.py3073 swap_memory=True)
/external/tensorflow/tensorflow/core/ops/
Dops.pbtxt27662 name: "swap_memory"
27688 name: "swap_memory"
/external/tensorflow/tensorflow/core/ops/compat/
Dops_history.v1.pbtxt60720 name: "swap_memory"
60746 name: "swap_memory"
Dops_history.v0.pbtxt35505 name: "swap_memory"