Searched refs:keep_input (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/training/ |
| D | input.py | 478 def _store_sparse_tensors(tensor_list, enqueue_many, keep_input, argument
527 def _maybe_store_sparse(t, map_op_name, keep_input): argument
530 keep_input,
533 def _maybe_store_many_sparse(t, map_op_name, keep_input): argument
536 keep_input,
541 def _sparse_values_to_keep(t, keep_input): argument
546 return array_ops.gather(keep_input, row_values)
547 if keep_input.shape.ndims == 1:
548 t = sparse_ops.sparse_retain(t, _sparse_values_to_keep(t, keep_input))
554 return store_f(t, map_op_name, keep_input)
[all …]
|
| D | input_test.py | 830 keep_input = array_ops.squeeze(
834 keep_input,
890 keep_input=constant_op.constant([True, False]),
896 keep_input=constant_op.constant([[True], [False]]),
903 keep_input=array_ops.placeholder(dtypes.bool),
912 batched = inp.maybe_batch([sparse], keep_input=True, batch_size=2)
921 keep_input=True,
932 keep_input=[True, False],
944 batched = inp.maybe_batch([sparse], keep_input=True, batch_size=2)
955 keep_input=True,
[all …]
|
| /external/perfetto/tools/ |
| D | diff_test_trace_processor.py | 164 keep_input, rebase): argument
228 if keep_input:
418 args.keep_input, args.rebase)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.train.pbtxt | 385 …argspec: "args=[\'tensors\', \'keep_input\', \'batch_size\', \'num_threads\', \'capacity\', \'enqu…
389 …argspec: "args=[\'tensors_list\', \'keep_input\', \'batch_size\', \'capacity\', \'enqueue_many\', …
393 …argspec: "args=[\'tensors\', \'batch_size\', \'capacity\', \'min_after_dequeue\', \'keep_input\', …
397 …tensors_list\', \'batch_size\', \'capacity\', \'min_after_dequeue\', \'keep_input\', \'seed\', \'e…
|