Searched defs:swap_memory (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | functional_ops.py | 51 swap_memory=False, argument
179 swap_memory=False, argument
247 swap_memory=False, argument
376 swap_memory=False, argument
444 swap_memory=False, argument
709 swap_memory=False, argument
|
| H A D | rnn.py | 354 swap_memory=False, argument
510 swap_memory=False, argument
769 swap_memory, argument
986 swap_memory=False, argument
|
| H A D | map_fn.py | 49 swap_memory=False, argument
633 swap_memory=False, argument
|
| H A D | control_flow_ops.py | 1418 swap_memory=False, argument
1447 swap_memory, name): argument
1551 def swap_memory(self): member in WhileContext
2353 swap_memory=False, argument
2534 swap_memory=False, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| H A D | rnn_test.py | 623 def _dynamic_rnn_swap_memory_benchmark(inputs_t, sequence_length, swap_memory): argument
681 swap_memory, nn): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/autograph/lang/ |
| H A D | directives.py | 48 swap_memory=UNSPECIFIED, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_map_ops.py | 32 swap_memory=False, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/ |
| H A D | control_flow.proto | 64 bool swap_memory = 4; field
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/risc/ |
| H A D | risc_ops.py | 270 swap_memory=False, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | def_function_test.py | 389 swap_memory=False): argument
|