Searched refs:keep_input (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/training/ |
| D | input.py | 479 def _store_sparse_tensors(tensor_list, enqueue_many, keep_input, argument
528 def _maybe_store_sparse(t, map_op_name, keep_input): argument
531 keep_input,
534 def _maybe_store_many_sparse(t, map_op_name, keep_input): argument
537 keep_input,
542 def _sparse_values_to_keep(t, keep_input): argument
547 return array_ops.gather(keep_input, row_values)
548 if keep_input.shape.ndims == 1:
549 t = sparse_ops.sparse_retain(t, _sparse_values_to_keep(t, keep_input))
555 return store_f(t, map_op_name, keep_input)
[all …]
|
| D | input_test.py | 852 keep_input = array_ops.squeeze(
856 keep_input,
905 keep_input=constant_op.constant([True, False]),
911 keep_input=constant_op.constant([[True], [False]]),
918 keep_input=array_ops.placeholder(dtypes.bool),
927 batched = inp.maybe_batch([sparse], keep_input=True, batch_size=2)
936 [sparse], keep_input=True, batch_size=2, enqueue_many=True)
945 [sparse], keep_input=[True, False], batch_size=2, enqueue_many=True)
955 batched = inp.maybe_batch([sparse], keep_input=True, batch_size=2)
966 [sparse], keep_input=True, batch_size=2, enqueue_many=True)
[all …]
|
| /external/tensorflow/tensorflow/contrib/training/python/training/ |
| D | bucket_ops.py | 73 keep_input=True, argument
179 keep_input = _validate_keep_input(keep_input, enqueue_many=False)
181 tensor_list, enqueue_many=False, keep_input=keep_input)
243 keep_input,
313 keep_input=True, argument
419 keep_input=keep_input,
|
| D | sampling_ops.py | 123 keep_input=random_ops.random_uniform([]) < cur_prob,
255 keep_input=random_ops.random_uniform([]) < cur_prob,
|
| D | bucket_ops_test.py | 276 keep_input = math_ops.equal(which_bucket, 0)
283 keep_input=keep_input,
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.train.pbtxt | 377 …argspec: "args=[\'tensors\', \'keep_input\', \'batch_size\', \'num_threads\', \'capacity\', \'enqu…
381 …argspec: "args=[\'tensors_list\', \'keep_input\', \'batch_size\', \'capacity\', \'enqueue_many\', …
385 …argspec: "args=[\'tensors\', \'batch_size\', \'capacity\', \'min_after_dequeue\', \'keep_input\', …
389 …tensors_list\', \'batch_size\', \'capacity\', \'min_after_dequeue\', \'keep_input\', \'seed\', \'e…
|
| /external/tensorflow/tensorflow/contrib/timeseries/python/timeseries/ |
| D | input_pipeline.py | 777 keep_input=non_decreasing,
|