STT-tensorflow/tensorflow/core/util
Yunxing Dai 41b3e84fa9 Support more cases in strided slice ops.
- When ranges are dynamic, we can lower strided slice ops into xla dynamic slices.
- Optimize xla buidler dynamism inference to not materialize parameter tuple.
- Export some helper data structures from strided_slice_op.

PiperOrigin-RevId: 356839276
Change-Id: I3b5d9fcb1289496029afbf97e49df49df26ce699
2021-02-10 15:38:32 -08:00
..
ctc Add target environment constraints to a subset of TensorFlow packages and targets. 2020-09-21 10:57:01 -07:00
proto Add target environment constraints to a subset of TensorFlow packages and targets. 2020-09-21 10:57:01 -07:00
sparse Update benchmarks to newer API 2021-02-09 13:12:44 -08:00
tensor_bundle fix OOMed issue due to tc-malloc fragmentation. 2021-01-20 17:43:11 -08:00
activation_mode.cc
activation_mode.h
batch_util.cc Rollforward: Add uint32 & uint64 to TF_CALL_INTEGRAL_TYPES 2020-06-22 23:09:59 -07:00
batch_util.h add a tensorflow::batch_util::CopyContiguousSlices utility function for 2020-05-27 10:26:55 -07:00
bcast_test.cc Fix undefined behavior in BCast. 2021-01-27 12:35:18 -08:00
bcast.cc
bcast.h Fix undefined behavior in BCast. 2021-01-27 12:35:18 -08:00
BUILD [NFC][Stack Traces] Do not build a functor where we can use the translation map directly 2021-02-09 10:18:15 -08:00
command_line_flags_test.cc minor spelling tweaks 2020-02-27 15:42:16 +09:00
command_line_flags.cc Update tf_mlir_enable_mlir_bridge support unspecified 2020-10-16 09:55:12 -07:00
command_line_flags.h Update tf_mlir_enable_mlir_bridge support unspecified 2020-10-16 09:55:12 -07:00
cuda_solvers.cc Break up core/kernels/BUILD (part 1 of N): 2020-08-04 17:24:28 -07:00
cuda_solvers.h Break up core/kernels/BUILD (part 1 of N): 2020-08-04 17:24:28 -07:00
cuda_sparse.cc Break up core/kernels/BUILD (part 1 of N): 2020-08-04 17:24:28 -07:00
cuda_sparse.h Qualify uses of std::string 2020-09-17 19:45:10 -07:00
debug_events_writer_test.cc [tfdbg2] Add tfdbg_run_id to metadata of data dumps 2020-06-15 13:52:52 -07:00
debug_events_writer.cc [DebuggerV2] Enable debug_v2_ops_test & debug_events_writer_test on Windows 2020-12-30 09:50:40 -08:00
debug_events_writer.h [tfdbg2] Add tfdbg_run_id to metadata of data dumps 2020-06-15 13:52:52 -07:00
device_name_utils_test.cc internal tests cleanup 2020-11-03 19:22:55 -08:00
device_name_utils.cc Make multiple changes to enable TPU single host training e2e. The changes include: 2021-01-26 17:05:42 -08:00
device_name_utils.h Introduce helper ParseFullOrLocalName 2020-09-02 14:21:04 -07:00
dump_graph_test.cc Refactor dump_graph to work on WriteableFile internally 2020-04-13 15:48:24 -07:00
dump_graph.cc Merge pull request #44794 from xinan-jiang:pr/cost-model 2021-02-05 10:46:31 -08:00
dump_graph.h Merge pull request #44794 from xinan-jiang:pr/cost-model 2021-02-05 10:46:31 -08:00
einsum_op_util.cc
einsum_op_util.h
env_var.cc Split out env_var from :lib target. 2020-01-17 17:22:04 -08:00
env_var.h Qualifying "string" as std::string 2021-02-05 16:16:25 -08:00
equal_graph_def_test.cc
equal_graph_def.cc
equal_graph_def.h
event.proto Add go_package options to more proto files 2020-09-14 14:11:15 -07:00
events_writer_test.cc
events_writer.cc Use explicit primitive types with strings::Printf 2020-03-08 20:58:24 -07:00
events_writer.h Qualify uses of std::string 2020-08-12 09:57:36 -07:00
example_proto_fast_parsing_test.cc
example_proto_fast_parsing_test.proto
example_proto_fast_parsing.cc Fixed bug where Tensorflow misreports sizes when sizes are mismatched. 2020-10-08 16:18:10 -07:00
example_proto_fast_parsing.h minor spelling tweaks 2020-02-27 15:42:16 +09:00
example_proto_helper_test.cc
example_proto_helper.cc Prevent dereferencing nullptr in CopyIntoSparseTensor called with empty tensor. 2020-07-28 14:20:33 -07:00
example_proto_helper.h
exec_on_stall_test.cc
exec_on_stall.h Prefixing TensorFlow thread annotation macros with TF_. 2020-03-05 08:42:01 -08:00
gpu_cuda_alias.h
gpu_device_functions.h Add param name comments 2020-11-05 07:39:46 -08:00
gpu_kernel_helper_test.cu.cc Merge pull request #40305 from ROCmSoftwarePlatform:google-upstream-kernel-helper-test 2020-06-22 11:51:48 -07:00
gpu_kernel_helper.h
gpu_launch_config.h PR #44471: [ROCm] Update to use ROCm 3.9 (when building TF with --config=rocm) 2020-11-10 00:57:26 -08:00
guarded_philox_random.cc
guarded_philox_random.h Prefixing TensorFlow thread annotation macros with TF_. 2020-03-05 08:42:01 -08:00
image_resizer_state.h Merge pull request #39243 from duncanriach:deterministic-image-resize-bilinear 2020-09-22 08:54:54 -07:00
incremental_barrier_test.cc In tensorflow/core/util/, introduce a IncrementalBarrier library. 2020-05-26 12:56:05 -07:00
incremental_barrier.cc In tensorflow/core/util/, introduce a IncrementalBarrier library. 2020-05-26 12:56:05 -07:00
incremental_barrier.h In tensorflow/core/util/, introduce a IncrementalBarrier library. 2020-05-26 12:56:05 -07:00
managed_stack_trace.h [NFC][Stack Traces] Do not build a functor where we can use the translation map directly 2021-02-09 10:18:15 -08:00
matmul_autotune.cc Rollback PR #43237: Integrate cuBLASLt API into backend 2020-10-28 17:04:53 -07:00
matmul_autotune.h Rollback PR #43237: Integrate cuBLASLt API into backend 2020-10-28 17:04:53 -07:00
matmul_bcast_test.cc Fix regression in (non-batch) matrix multiplication: 2020-11-17 15:58:05 -08:00
matmul_bcast.h Fix regression in (non-batch) matrix multiplication: 2020-11-17 15:58:05 -08:00
memmapped_file_system_test.cc
memmapped_file_system_writer.cc
memmapped_file_system_writer.h
memmapped_file_system.cc Moving rest of the filesystems to Transactional API 2020-08-04 18:04:52 -07:00
memmapped_file_system.h Moving rest of the filesystems to Transactional API 2020-08-04 18:04:52 -07:00
memmapped_file_system.proto
mirror_pad_mode.cc
mirror_pad_mode.h
mkl_threadpool.h minor bug fixes for unit tests. 2020-05-27 14:48:52 -07:00
mkl_types.h Bringing back PR #42339. 2020-09-11 13:21:53 -07:00
mkl_util_test.cc MKL DNN 0.x code cleanup - MKLSlice op 2020-09-11 09:28:38 -07:00
mkl_util.h Fix clang compilation errors for --config=mkl_threadpool. 2020-12-28 18:42:21 -08:00
overflow_test.cc Make MultiplyWithoutOverflow be a total function. 2020-11-13 09:47:06 -08:00
overflow.h Make MultiplyWithoutOverflow be a total function. 2020-11-13 09:47:06 -08:00
padding.cc Update tensorflow/core/util/padding.cc 2020-07-28 20:59:14 -04:00
padding.h Qualify uses of std::string 2020-07-31 10:27:13 -07:00
permutation_input_iterator.h
permutation_output_iterator.h
port.cc [TF XLA] Add ability to convert SavedModel subgraphs to compiled [XLA CPU] objects via saved_model_cli. 2020-01-16 14:26:12 -08:00
port.h [TF XLA] Add ability to convert SavedModel subgraphs to compiled [XLA CPU] objects via saved_model_cli. 2020-01-16 14:26:12 -08:00
presized_cuckoo_map_test.cc internal tests cleanup 2020-11-03 19:22:55 -08:00
presized_cuckoo_map.h
ptr_util.h
ragged_to_dense_util_common.h Fixing build of when TFLite and TF are linked in one binary. 2020-11-30 16:07:44 -08:00
ragged_to_dense_util_test.cc Add RaggedTensorToTensor to flex delegate 2020-07-12 20:32:45 -07:00
ragged_to_dense_util.cc Moving common with TFLite ragged tensor code to a separate file. 2020-09-02 11:43:03 -07:00
ragged_to_dense_util.h Moving common with TFLite ragged tensor code to a separate file. 2020-09-02 11:43:03 -07:00
reffed_status_callback_test.cc
reffed_status_callback.h Prefixing TensorFlow thread annotation macros with TF_. 2020-03-05 08:42:01 -08:00
reporter_test.cc Add a method to add metric to the benchmark entry. 2020-08-09 19:38:02 -07:00
reporter.cc Add a method to add metric to the benchmark entry. 2020-08-09 19:38:02 -07:00
reporter.h Add a method to add metric to the benchmark entry. 2020-08-09 19:38:02 -07:00
rocm_solvers.cc Break up core/kernels/BUILD (part 1 of N): 2020-08-04 17:24:28 -07:00
rocm_solvers.h Break up core/kernels/BUILD (part 1 of N): 2020-08-04 17:24:28 -07:00
rocm_sparse.cc [ROCm] Updates to dynamically load the ROCm "hipsparse" library 2020-08-06 22:28:25 +00:00
saved_tensor_slice_util_test.cc
saved_tensor_slice_util.cc
saved_tensor_slice_util.h Rollforward: Add uint32 & uint64 to TF_CALL_INTEGRAL_TYPES 2020-06-22 23:09:59 -07:00
saved_tensor_slice.proto
semver_test.cc
stat_summarizer_options.h Update stats_calculator to allow dumping values as CSV 2020-02-07 11:31:55 -08:00
stat_summarizer_test.cc
stat_summarizer.cc Optimize calls to std::string::find() and friends for a single char. 2020-12-17 18:11:13 -08:00
stat_summarizer.h
stats_calculator_test.cc
stats_calculator.cc Update stats_calculator to fix csv format error in node name 2020-02-12 19:22:57 -08:00
stats_calculator.h
stream_executor_util.h
strided_slice_op.cc Support more cases in strided slice ops. 2021-02-10 15:38:32 -08:00
strided_slice_op.h Support more cases in strided slice ops. 2021-02-10 15:38:32 -08:00
tensor_format_test.cc
tensor_format.cc Use string_view type for FormatFromString and FilterFormatFromString arguments 2020-08-07 00:41:05 -07:00
tensor_format.h Use string_view type for FormatFromString and FilterFormatFromString arguments 2020-08-07 00:41:05 -07:00
tensor_ops_util.h
tensor_slice_reader_cache.cc This CL optimizes C++11 range-based for loops where the variable is copied in each iteration but it would suffice to obtain it by const reference. This is only applied to loop variables of types that are expensive to copy which means they are not trivially copyable or have a non-trivial copy constructor or destructor. 2020-04-06 19:54:28 -07:00
tensor_slice_reader_cache.h
tensor_slice_reader_test.cc
tensor_slice_reader.cc This CL optimizes C++11 range-based for loops where the variable is copied in each iteration but it would suffice to obtain it by const reference. This is only applied to loop variables of types that are expensive to copy which means they are not trivially copyable or have a non-trivial copy constructor or destructor. 2020-04-06 19:54:28 -07:00
tensor_slice_reader.h Some const declarations changed to constexpr 2020-04-21 10:21:10 -07:00
tensor_slice_set_test.cc Update benchmarks to newer API 2021-02-09 13:12:44 -08:00
tensor_slice_set.cc
tensor_slice_set.h
tensor_slice_util_test.cc
tensor_slice_util.h
tensor_slice_writer_test.cc
tensor_slice_writer.cc
tensor_slice_writer.h Some const declarations changed to constexpr 2020-04-21 10:21:10 -07:00
test_log.proto Add IOS_BENCHMARK enum value to BenchmarkType 2020-05-14 00:15:02 -07:00
transform_output_iterator.h
use_cudnn.cc Remove CanUseCudnn and CudnnConvComputeMode functions. 2020-08-18 12:04:59 -07:00
use_cudnn.h Remove CanUseCudnn and CudnnConvComputeMode functions. 2020-08-18 12:04:59 -07:00
util.cc
util.h Qualify uses of std::string 2020-07-31 10:27:13 -07:00
work_sharder_test.cc internal tests cleanup 2020-11-03 19:22:55 -08:00
work_sharder.cc
work_sharder.h
xla_config_registry.cc Prefixing TensorFlow thread annotation macros with TF_. 2020-03-05 08:42:01 -08:00
xla_config_registry.h