experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
STT-tensorflow/tensorflow/compiler/jit
History
Sanjoy Das d7634bbfaf Finish migrating {Tf|Platform}GpuId to {Tf|Platform}DeviceId 
PiperOrigin-RevId: 361252995
Change-Id: I818798fc00efe7b98c35145ce067204d9e023895
2021-03-05 17:15:52 -08:00
..
kernels
[TF2XLA] Support must-be-constant resource variables for compilation
2020-11-09 15:04:58 -08:00
ops
PY2 removal cleanup
2021-01-15 16:48:57 -08:00
tests
Add target environment constraints to a subset of TensorFlow packages and targets.
2020-09-21 10:57:01 -07:00
BUILD
Define helper function for effective MlirBridgeRollout enum
2021-03-02 05:29:23 -08:00
build_xla_ops_pass_test.cc
Use StatefulPartitionedCall instead of PartitionedCall in auto-clustering
2021-02-16 12:00:23 -08:00
build_xla_ops_pass.cc
Use StatefulPartitionedCall instead of PartitionedCall in auto-clustering
2021-02-16 12:00:23 -08:00
build_xla_ops_pass.h
…
clone_constants_for_better_clustering_test.cc
Make constant cloning more aggressive (i.e. make it clone more often)
2019-10-07 12:23:37 -07:00
clone_constants_for_better_clustering.cc
…
clone_constants_for_better_clustering.h
…
cluster_scoping_pass_test.cc
[Build cleanup] Update #includes of moved header "graph/graph_constructor.h".
2020-04-29 09:20:48 -07:00
cluster_scoping_pass.cc
Merge pull request #31594 from trentlo:github-cluster-scoping
2019-08-21 20:15:13 -07:00
cluster_scoping_pass.h
Merge pull request #31594 from trentlo:github-cluster-scoping
2019-08-21 20:15:13 -07:00
compilability_check_util_test.cc
[TF2XLA] Check 'case' op compilation from autoclustering codepath, but not from tf.function
2020-10-07 12:44:29 -07:00
compilability_check_util.cc
[TF2XLA] Remove redundant compilability checking in xla_kernel_creator
2021-02-22 14:58:08 -08:00
compilability_check_util.h
[TF2XLA] Remove redundant compilability checking in xla_kernel_creator
2021-02-22 14:58:08 -08:00
compilation_passes_test_main.cc
…
deadness_analysis_internal.h
…
deadness_analysis_test.cc
[Build cleanup] Update #includes of moved header "graph/graph_constructor.h".
2020-04-29 09:20:48 -07:00
deadness_analysis.cc
…
deadness_analysis.h
…
defs.cc
[TF2XLA] Add a utility function to check whether GraphDef can trigger XLA compilation
2020-08-28 17:11:30 -07:00
defs.h
[TF2XLA] Add a utility function to check whether GraphDef can trigger XLA compilation
2020-08-28 17:11:30 -07:00
device_util_test.cc
[XLA] Allow multiple compatible devices in device_util
2019-07-12 18:47:39 -07:00
device_util.cc
c, compiler, jit resolutions
2020-07-26 20:42:00 +00:00
device_util.h
Merge pull request #41734 from tg-at-google:wsign-compare-semi-final-tf-c-compiler-jit
2020-07-30 13:30:14 -07:00
encapsulate_subgraphs_pass_test.cc
Adjust TPU build dependencies.
2020-07-27 15:37:03 -07:00
encapsulate_subgraphs_pass.cc
[XLA] Insert control edges from write to read instructions for same buffers inside loops
2020-10-26 17:30:57 -07:00
encapsulate_subgraphs_pass.h
[NFC] Clarify documentation of JIT passes
2019-12-19 11:20:28 -08:00
encapsulate_util_test.cc
…
encapsulate_util.cc
c, compiler, jit resolutions
2020-07-26 20:42:00 +00:00
encapsulate_util.h
minor spelling tweaks
2019-12-09 18:21:12 +09:00
encapsulate_xla_computations_pass_test.cc
[TF2XLA] Add a utility function to check whether GraphDef can trigger XLA compilation
2020-08-28 17:11:30 -07:00
encapsulate_xla_computations_pass.cc
Use TryGetNodeAttr instead of GetNodeAttr in a few places
2021-02-15 14:26:06 -08:00
encapsulate_xla_computations_pass.h
[TF2XLA] Add a utility function to check whether GraphDef can trigger XLA compilation
2020-08-28 17:11:30 -07:00
extract_outside_compilation_pass_test.cc
Adjust TPU build dependencies.
2020-07-27 15:37:03 -07:00
extract_outside_compilation_pass.cc
Fix outside compilation to add new function instead of inplace function update.
2021-01-08 17:16:29 -08:00
extract_outside_compilation_pass.h
Support summary in while loop.
2019-06-28 12:56:25 -07:00
flags.cc
Define helper function for effective MlirBridgeRollout enum
2021-03-02 05:29:23 -08:00
flags.h
Define helper function for effective MlirBridgeRollout enum
2021-03-02 05:29:23 -08:00
force_xla_constants_on_host_pass_test.cc
[Build cleanup] Update #includes of moved header "graph/graph_constructor.h".
2020-04-29 09:20:48 -07:00
force_xla_constants_on_host_pass.cc
[TF2XLA] [NFC] Simplify GetBodyAndConstantsAndResources
2020-09-02 22:03:20 -07:00
force_xla_constants_on_host_pass.h
[TF/XLA] Force all tensors which need to be constant during the XLA compilation to be located on the host
2020-04-01 11:34:32 -07:00
get_compiler_ir.cc
Extend experimental_get_compiler_ir to get serialized HLO modules.
2021-01-26 23:30:30 -08:00
get_compiler_ir.h
Extend experimental_get_compiler_ir to get serialized HLO modules.
2021-01-26 23:30:30 -08:00
increase_dynamism_for_auto_jit_pass_test.cc
…
increase_dynamism_for_auto_jit_pass.cc
c, compiler, jit resolutions
2020-07-26 20:42:00 +00:00
increase_dynamism_for_auto_jit_pass.h
…
introduce_floating_point_jitter_pass_internal.h
Add a debug-only pass that introduces a small error to a designated TF node
2019-03-28 12:25:47 -07:00
introduce_floating_point_jitter_pass_test.cc
…
introduce_floating_point_jitter_pass.cc
…
introduce_floating_point_jitter_pass.h
Add a debug-only pass that introduces a small error to a designated TF node
2019-03-28 12:25:47 -07:00
jit_compilation_pass_registration.cc
[TF/XLA] Force all tensors which need to be constant during the XLA compilation to be located on the host
2020-04-01 11:34:32 -07:00
mark_for_compilation_pass_test_helper.cc
Add a new unittest in mark_for_compilation_pass_test.
2019-08-16 13:37:19 -07:00
mark_for_compilation_pass_test_helper.h
…
mark_for_compilation_pass_test.cc
Rollback of rollback of disabling XLA:CPU/GPU devices by default.
2020-08-11 13:06:59 -07:00
mark_for_compilation_pass.cc
Introduce Xla{Dot,Conv}V2
2021-02-22 11:25:16 -08:00
mark_for_compilation_pass.h
[TF2XLA] [NFC] Make a function only used in a single library fileprivate
2020-09-18 23:30:47 -07:00
node_matchers_test.cc
…
node_matchers.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-07 14:38:02 -07:00
node_matchers.h
…
partially_decluster_pass_test.cc
Rollback of rollback of disabling XLA:CPU/GPU devices by default.
2020-08-11 13:06:59 -07:00
partially_decluster_pass.cc
…
partially_decluster_pass.h
Decluster some must-be-constant ops to reduce XLA recompilations
2018-09-07 18:53:18 -07:00
rearrange_function_argument_pass_test.cc
Update RearrangeFunctionArgumentForFunctionTest to use correct index when creating Retval nodes.
2019-09-27 14:25:28 -07:00
report_clustering_info_pass.cc
Roll forward "Add an XLA "activity listener" mechanism."
2019-06-14 17:14:19 -07:00
report_clustering_info_pass.h
…
resource_operation_safety_analysis_test.cc
[Build cleanup] Split "core_cpu_impl" into fine-grained targets (4/n).
2020-04-28 09:53:59 -07:00
resource_operation_safety_analysis.cc
Storing kRead kinds doesn't have any effect, but might take a lot of RAM.
2020-03-17 09:54:28 -07:00
resource_operation_safety_analysis.h
[XLA] Insert control edges from write to read instructions for same buffers inside loops
2020-10-26 17:30:57 -07:00
shape_inference_helpers.cc
…
shape_inference_helpers.h
…
shape_inference_test.cc
…
shape_inference.cc
c, compiler, jit resolutions
2020-07-26 20:42:00 +00:00
shape_inference.h
…
test_util.cc
…
test_util.h
…
xla_activity_listener_test.cc
…
xla_activity_listener.cc
…
xla_activity_listener.h
…
xla_activity_logging_listener.cc
…
xla_activity.proto
[TF:XLA] Logging of known cases of performance deficiencies using XLA with Tensorflow.
2019-08-07 04:04:01 -07:00
xla_cluster_util_test.cc
[Build cleanup] Update #includes of moved header "graph/graph_constructor.h".
2020-04-29 09:20:48 -07:00
xla_cluster_util.cc
c, compiler, jit resolutions
2020-07-26 20:42:00 +00:00
xla_cluster_util.h
[XLA] Insert control edges from write to read instructions for same buffers inside loops
2020-10-26 17:30:57 -07:00
xla_compilation_cache_test.cc
Internal tests cleanup.
2020-10-27 10:16:12 -07:00
xla_compilation_cache.cc
Lookup functions in optional FunctionLibraryDefinition when checking for supported features for MLIR bridge.
2021-03-05 09:34:14 -08:00
xla_compilation_cache.h
Add MlirXlaOpKernel, which is used to implement XlaOpKernels using MLIR legalization.
2021-03-03 07:27:30 -08:00
xla_compile_on_demand_op.cc
[TF2XLA] Support must-be-constant resource variables for compilation
2020-11-09 15:04:58 -08:00
xla_compile_on_demand_op.h
[TF2XLA] [NFC] Simplify XlaPlatformInfoFromDevice
2020-08-28 11:20:02 -07:00
xla_cpu_device.cc
[NFC] [TF2XLA] Reduce log spam: no need to say every time that we are not creating XLA devices
2020-12-09 21:14:55 -08:00
xla_device_context.cc
Merge pull request #43134 from ROCmSoftwarePlatform:google-upstream-thenexecute-deadlock
2020-09-18 05:20:22 -07:00
xla_device_context.h
Merge pull request #43134 from ROCmSoftwarePlatform:google-upstream-thenexecute-deadlock
2020-09-18 05:20:22 -07:00
xla_device_ops.cc
Rollback of rollback of [TF/XLA] Enable input/output aliasing in the TF2XLA bridge
2020-07-17 15:17:25 -07:00
xla_device_ops.h
Register VariableShapeOp for TPU.
2021-02-08 15:50:16 -08:00
xla_device.cc
Rationale:
2020-11-03 12:55:59 -08:00
xla_device.h
[TF2XLA] [NFC] Simplify XlaPlatformInfoFromDevice
2020-08-28 11:20:02 -07:00
xla_gpu_device.cc
Finish migrating {Tf|Platform}GpuId to {Tf|Platform}DeviceId
2021-03-05 17:15:52 -08:00
xla_kernel_creator_test.cc
…
xla_kernel_creator.cc
[TF2XLA] Remove redundant compilability checking in xla_kernel_creator
2021-02-22 14:58:08 -08:00
xla_kernel_creator.h
Replace NodeDef with std::shared_ptr<NodeProperties> in the kernel creation code paths and try to avoid as many copies of NodeDefs as possible. This will in most cases allow sharing the NodeDef between the OpKernel and the graph Node from which it is created.
2020-02-18 13:20:06 -08:00
xla_launch_util.cc
Propagate the frame where the resource var was created to resource var handles
2021-01-25 14:25:35 -08:00
xla_launch_util.h
Implement allocation tracking support for XLA allocated tensors.
2021-03-05 02:35:46 -08:00
xla_ops_on_regular_devices.cc
Introduce Xla{Dot,Conv}V2
2021-02-22 11:25:16 -08:00
xla_platform_info.cc
[TF2XLA] [NFC] Simplify GenerateCompilerOptions and GetAllocator
2020-08-28 11:34:15 -07:00
xla_platform_info.h
[TF2XLA] [NFC] Simplify GenerateCompilerOptions and GetAllocator
2020-08-28 11:34:15 -07:00
xla_tensor.cc
[TF/XLA] Remove the wrapping function XlaTensor::RefCountIsOne
2020-06-15 11:44:06 -07:00
xla_tensor.h
[TF2XLA] Remove XlaTensor::set_host_tensor. It creates unnecessary complication
2020-09-02 17:02:00 -07:00
xla_tpu_device.cc
[TPU] Move TPU node and system device initializers to compiler/jit
2021-01-19 22:43:21 -08:00
xla_tpu_device.h
[TPU] Move TPU node and system device initializers to compiler/jit
2021-01-19 22:43:21 -08:00