| .. |
|
experimental
|
Update "master" to "dispatch"/"dispatcher" in tf.data service terminology.
|
2020-07-16 13:07:28 -07:00 |
|
batch_dataset_op_test.cc
|
|
|
|
batch_dataset_op.cc
|
Limit reserve size by 2**16 in dataset batch op when drop_remainder is false.
|
2020-06-16 13:04:55 -07:00 |
|
batch_dataset_op.h
|
|
|
|
BUILD
|
[tf.data] Refactoring of tf.data static optimizations.
|
2020-05-27 14:50:23 -07:00 |
|
cache_dataset_ops_test.cc
|
[tf.data] Internal cleanup.
|
2020-02-18 11:46:51 -08:00 |
|
cache_dataset_ops.cc
|
[tf.data] Improvements to the performance modeling framework.
|
2020-06-03 12:02:15 -07:00 |
|
cache_dataset_ops.h
|
[tf.data] This CL changes how the in-memory cache resource is managed, making it possible for the cache dataset to support both a) sharing of the cache across iterators and b) serialization. As a consequence, this CL enables sharing of the cache across iterators for tf.distribute and tf.data service use cases (which require serialization support).
|
2020-04-20 13:55:08 -07:00 |
|
cache_ops.cc
|
[tf.data] Memory-safe implementation of sharing access to the memory cache.
|
2020-04-21 20:47:38 -07:00 |
|
cache_ops.h
|
[tf.data] Memory-safe implementation of sharing access to the memory cache.
|
2020-04-21 20:47:38 -07:00 |
|
captured_function.cc
|
Support packed variable for tf data captured function.
|
2020-06-22 16:24:37 -07:00 |
|
captured_function.h
|
[tf.data] Cleanup of C++ user-defined function execution utilities.
|
2020-06-11 16:24:57 -07:00 |
|
concatenate_dataset_op_test.cc
|
|
|
|
concatenate_dataset_op.cc
|
[tf.data] Adding support for overriding external state policy for checkpointing.
|
2020-03-17 13:44:46 -07:00 |
|
concatenate_dataset_op.h
|
|
|
|
dataset_ops.cc
|
[tf.data] Moving cardinality to core API.
|
2020-05-01 14:12:38 -07:00 |
|
dataset_ops.h
|
[tf.data] Moving cardinality to core API.
|
2020-05-01 14:12:38 -07:00 |
|
dataset_test_base.cc
|
Rollback: Add uint32 & uint64 to TF_CALL_INTEGRAL_TYPES
|
2020-06-20 15:17:39 -07:00 |
|
dataset_test_base.h
|
[Build cleanup] Update #includes of moved header "graph/graph_constructor.h".
|
2020-04-29 09:20:48 -07:00 |
|
dataset_utils_test.cc
|
Enable op-level dataset determinism configuration.
|
2020-02-03 16:07:36 -08:00 |
|
dataset_utils.cc
|
Merge pull request #40118 from zhuzilin:iterator-next-check
|
2020-06-05 12:08:01 -07:00 |
|
dataset_utils.h
|
refactor VerifyTypesMatch and VerifyShapeCompatible
|
2020-06-05 12:15:48 +08:00 |
|
filter_dataset_op_test.cc
|
|
|
|
filter_dataset_op.cc
|
[tf.data] Adding support for overriding external state policy for checkpointing.
|
2020-03-17 13:44:46 -07:00 |
|
filter_dataset_op.h
|
|
|
|
fixed_length_record_dataset_op_test.cc
|
Use Env::LocalTempFilename for a temp filename.
|
2020-02-20 10:27:38 -08:00 |
|
fixed_length_record_dataset_op.cc
|
[tf.data] Avoid contention in bytes-read metric collection.
|
2020-03-20 12:40:40 -07:00 |
|
fixed_length_record_dataset_op.h
|
|
|
|
flat_map_dataset_op_test.cc
|
|
|
|
flat_map_dataset_op.cc
|
[tf.data] Adding support for overriding external state policy for checkpointing.
|
2020-03-17 13:44:46 -07:00 |
|
flat_map_dataset_op.h
|
|
|
|
generator_dataset_op.cc
|
[tf.data] Adding support for overriding external state policy for checkpointing.
|
2020-03-17 13:44:46 -07:00 |
|
generator_dataset_op.h
|
|
|
|
interleave_dataset_op_test.cc
|
|
|
|
interleave_dataset_op.cc
|
[tf.data] Adding support for overriding external state policy for checkpointing.
|
2020-03-17 13:44:46 -07:00 |
|
interleave_dataset_op.h
|
|
|
|
iterator_ops.cc
|
refactor VerifyTypesMatch and VerifyShapeCompatible
|
2020-06-05 12:15:48 +08:00 |
|
iterator_ops.h
|
add shape and type check for IteratorGetNextOp and ToSingleElementOp
|
2020-06-03 17:13:00 +08:00 |
|
map_dataset_op_test.cc
|
use kUnknownCardinality in test
|
2020-06-07 08:46:15 +08:00 |
|
map_dataset_op.cc
|
fix code style
|
2020-06-05 14:04:16 +08:00 |
|
map_dataset_op.h
|
|
|
|
map_defun_op_test.cc
|
|
|
|
map_defun_op.cc
|
Switch CallFrameInterface::GetArg(int, Tensor*) to take (int, const Tensor**).
|
2020-04-10 18:51:55 -07:00 |
|
map_defun_op.h
|
|
|
|
model_dataset_op.cc
|
[tf.data] Adding a metric for bytes produced and consumed by individual transformations, refactoring infrastructure for recording tf.data metrics, and moving the metrics API and implementation from common_runtime to framework.
|
2020-04-06 10:39:03 -07:00 |
|
multi_device_iterator_ops.cc
|
[tf.data] Addressing TODOs regarding deprecated APIs.
|
2020-03-06 17:22:31 -08:00 |
|
name_utils_test.cc
|
|
|
|
name_utils.cc
|
|
|
|
name_utils.h
|
|
|
|
optimize_dataset_op_mobile.cc
|
|
|
|
optimize_dataset_op_test.cc
|
|
|
|
optimize_dataset_op.cc
|
Handle DeadlineExceeded in OptimizeDatasetOp::MakeDataset correctly. Before this change, if we encounter a DeadlineExceeded error, the output dataset will not be set (i.e. be a nullptr), resulting in "Read uninitialized Dataset variant" when we try to use it.
|
2020-06-05 15:20:31 -07:00 |
|
optimize_dataset_op.h
|
|
|
|
optional_ops.cc
|
|
|
|
optional_ops.cu.cc
|
|
|
|
optional_ops.h
|
|
|
|
padded_batch_dataset_op_test.cc
|
|
|
|
padded_batch_dataset_op.cc
|
[tf.data] Adding support for overriding external state policy for checkpointing.
|
2020-03-17 13:44:46 -07:00 |
|
padded_batch_dataset_op.h
|
|
|
|
parallel_interleave_dataset_op_test.cc
|
[tf.data] Fix some small typo in parallel_interleave_dataset_op.
|
2020-06-15 17:02:13 -07:00 |
|
parallel_interleave_dataset_op.cc
|
[tf.data] Fix some small typo in parallel_interleave_dataset_op.
|
2020-06-15 17:02:13 -07:00 |
|
parallel_interleave_dataset_op.h
|
Add API to control parallel interleave prefetching.
|
2020-02-12 15:09:49 -08:00 |
|
parallel_map_dataset_op_test.cc
|
use kUnknownCardinality in test
|
2020-06-07 08:46:15 +08:00 |
|
parallel_map_dataset_op.cc
|
fix code style
|
2020-06-05 14:04:16 +08:00 |
|
parallel_map_dataset_op.h
|
[tf.data] Fixing tracking of processing time for parallel map when inter-op parallelism is disabled.
|
2020-06-02 15:31:37 -07:00 |
|
prefetch_autotuner_test.cc
|
|
|
|
prefetch_autotuner.cc
|
segrating changes 2
|
2020-06-15 21:18:56 +00:00 |
|
prefetch_autotuner.h
|
|
|
|
prefetch_dataset_op_test.cc
|
|
|
|
prefetch_dataset_op.cc
|
[tf.data] Fix a bug in prefetch dataset serialization logic.
|
2020-05-27 13:36:42 -07:00 |
|
prefetch_dataset_op.h
|
|
|
|
random_seed_ops.cc
|
[tf.data] Make AnonymousSeedGeneratorHandleOp thread-safe.
|
2020-04-25 12:35:33 -07:00 |
|
random_seed_ops.h
|
[tf.data] Make AnonymousSeedGeneratorHandleOp thread-safe.
|
2020-04-25 12:35:33 -07:00 |
|
range_dataset_op_test.cc
|
|
|
|
range_dataset_op.cc
|
[tf.data] Adding support for overriding external state policy for checkpointing.
|
2020-03-17 13:44:46 -07:00 |
|
range_dataset_op.h
|
|
|
|
reduce_dataset_op_test.cc
|
|
|
|
repeat_dataset_op_test.cc
|
|
|
|
repeat_dataset_op.cc
|
[tf.data] Adding support for overriding external state policy for checkpointing.
|
2020-03-17 13:44:46 -07:00 |
|
repeat_dataset_op.h
|
|
|
|
rewrite_utils.cc
|
[Build cleanup] Update #includes of moved header "graph/graph_constructor.h".
|
2020-04-29 09:20:48 -07:00 |
|
rewrite_utils.h
|
[tf.data] Improvements to static optimizations.
|
2020-04-14 16:37:44 -07:00 |
|
serialization_utils.cc
|
|
|
|
serialization_utils.h
|
|
|
|
shard_dataset_op_test.cc
|
|
|
|
shard_dataset_op.cc
|
[tf.data] Adding support for overriding external state policy for checkpointing.
|
2020-03-17 13:44:46 -07:00 |
|
shard_dataset_op.h
|
|
|
|
shuffle_dataset_op_test.cc
|
[tf.data] Supporting shuffle + repeat fusion in TF 2 (and with correct reshuffle_each_iteration semantics).
|
2020-04-29 18:01:13 -07:00 |
|
shuffle_dataset_op.cc
|
Fix a key word typo in shuffle_dataset_op.
|
2020-06-04 11:56:05 -07:00 |
|
shuffle_dataset_op.h
|
[tf.data] Supporting shuffle + repeat fusion in TF 2 (and with correct reshuffle_each_iteration semantics).
|
2020-04-29 18:01:13 -07:00 |
|
single_threaded_executor_test.cc
|
Partial rollback of "[Single-threaded executor] Optimize the handling of OpKernel::const_tensor()."
|
2020-04-30 16:20:29 -07:00 |
|
single_threaded_executor.cc
|
[CallFrameInterface] Add methods ConsumeArg() and CanConsumeArg().
|
2020-05-05 11:09:16 -07:00 |
|
single_threaded_executor.h
|
|
|
|
skip_dataset_op_test.cc
|
|
|
|
skip_dataset_op.cc
|
[tf.data] Adding support for overriding external state policy for checkpointing.
|
2020-03-17 13:44:46 -07:00 |
|
skip_dataset_op.h
|
|
|
|
sparse_tensor_slice_dataset_op_test.cc
|
|
|
|
sparse_tensor_slice_dataset_op.cc
|
[tf.data] Adding support for overriding external state policy for checkpointing.
|
2020-03-17 13:44:46 -07:00 |
|
stats_utils.cc
|
|
|
|
stats_utils.h
|
|
|
|
take_dataset_op_test.cc
|
|
|
|
take_dataset_op.cc
|
[tf.data] Adding support for overriding external state policy for checkpointing.
|
2020-03-17 13:44:46 -07:00 |
|
take_dataset_op.h
|
|
|
|
tensor_dataset_op_test.cc
|
|
|
|
tensor_dataset_op.cc
|
[tf.data] Adding support for overriding external state policy for checkpointing.
|
2020-03-17 13:44:46 -07:00 |
|
tensor_dataset_op.h
|
|
|
|
tensor_slice_dataset_op_test.cc
|
|
|
|
tensor_slice_dataset_op.cc
|
[tf.data] Adding support for overriding external state policy for checkpointing.
|
2020-03-17 13:44:46 -07:00 |
|
tensor_slice_dataset_op.h
|
|
|
|
text_line_dataset_op_test.cc
|
Use Env::LocalTempFilename for a temp filename.
|
2020-02-20 10:24:21 -08:00 |
|
text_line_dataset_op.cc
|
[tf.data] Fix "passing result of std::move() as const reference argument" warning.
|
2020-03-20 17:07:29 -07:00 |
|
text_line_dataset_op.h
|
|
|
|
tf_record_dataset_op_test.cc
|
|
|
|
tf_record_dataset_op.cc
|
Merge pull request #39791 from rahul003:multipartdownload
|
2020-05-27 12:13:07 -07:00 |
|
tf_record_dataset_op.h
|
|
|
|
unbounded_thread_pool_test.cc
|
|
|
|
unbounded_thread_pool.cc
|
Update ResourceTagger implementation.
|
2020-02-28 15:53:18 -08:00 |
|
unbounded_thread_pool.h
|
Merge pull request #35640 from georgepaw:unbound_thread_queue_numa
|
2020-01-24 14:30:43 -08:00 |
|
window_dataset_op_test.cc
|
[tf.data] Addressing TODOs regarding deprecated APIs.
|
2020-03-06 17:22:31 -08:00 |
|
window_dataset_op.cc
|
[tf.data] Adding support for overriding external state policy for checkpointing.
|
2020-03-17 13:44:46 -07:00 |
|
window_dataset_op.h
|
|
|
|
window_dataset.cc
|
Fix tensorflow::errors:* calls, which use StrCat instead of StrFormat
|
2020-06-11 00:59:42 -07:00 |
|
window_dataset.h
|
|
|
|
zip_dataset_op_test.cc
|
|
|
|
zip_dataset_op.cc
|
[tf.data] Adding support for overriding external state policy for checkpointing.
|
2020-03-17 13:44:46 -07:00 |
|
zip_dataset_op.h
|
|
|
