| Commit message (Collapse) | Author | Age |
|
|
|
| |
PiperOrigin-RevId: 200795402
|
|
|
|
|
|
| |
This simply wraps the Transpose with a Conj.
PiperOrigin-RevId: 200793274
|
|
|
|
|
|
|
|
| |
The full list of inputs to the generated TF function is created by appending
the Tensor-valued keyword arguments (sorted by key) to the list of
Tensor-valued args.
PiperOrigin-RevId: 200792676
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
* EagerPyFunc now validates its assumption that returned tensors are backed by memory on the same device that the EagerPyFunc kernel executed on.
* Make the Python trampolining mechanism ensure that this requirement of the kernel is met.
* Allow tf.contrib.eager.py_func to execute correctly on devices other than CPU and GPU:0.
Prior to this change, tf.contrib.eager.py_func() would copy data from CPU to GPU:0 if necessary, but not the other way around. As a result, the assumptions made by the EagerPyFunc kernel implementation about the placement of returned tensors would be violated.
The test added in py_func_test.py, when executed on a machine with a GPU will:
- Fail with a segmentation fault (dereferencing GPU memory) without the changes to py_func.cc and script_ops.py
- Fail with an error message with the change to py_func.cc but without the change to script_ops.py
- Pass with changes to py_func.cc and script_ops.py
PiperOrigin-RevId: 200792057
|
|
|
|
| |
PiperOrigin-RevId: 200791799
|
|
|
|
| |
PiperOrigin-RevId: 200791012
|
|
|
|
| |
PiperOrigin-RevId: 200790410
|
|
|
|
|
|
|
|
| |
1) Update stats
2) Update the number of examples visited.
3) If the number of examples reaches the target, grow the tree.
PiperOrigin-RevId: 200790145
|
|
|
|
| |
PiperOrigin-RevId: 200789288
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
Prior this change, we were using the C API for everything except
Tensor.shape calls, which returned the result from the original Python
shape inference code. With this change, we use the C API in this case
as well. The C API has better shape inference, so this has the effect
of returning more precise shapes in some cases.
This change can be disabled by setting the environment variable
TF_C_API_GRAPH_CONSTRUCTION_SHAPES=0. However, this toggle will
be removed altogether in the near future.
This also fixes a bug in the SWIG that could cause large shape dimensions
to be incorrect.
PiperOrigin-RevId: 200783822
|
|
|
|
|
|
|
|
| |
This makes it more convenient to use layer of different dtypes in a model. Instead of having to manually cast intermediate tensors between layers of different dtypes, they will automatically be casted.
This is also useful for the upcoming mixed precision API.
PiperOrigin-RevId: 200783477
|
|
|
|
| |
PiperOrigin-RevId: 200783258
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
This change:
* Creates a new global variable, control_flow_ops._ENABLE_COND_V2, to use
cond_v2 by default when calling tf.cond. This variable can also be
controlled via the environment variable TF_ENABLE_COND_V2.
* Moves cond_v2 out of contrib so it's accessible from control_flow_ops.py.
* Lazily "imports" some modules in cond_v2 to avoid circular dependencies.
Note that these lazy "imports" must be imported by the cond_v2 caller (or
recursively by one of the caller's imports) in order for cond_v2 to have
access to them.
* Renames the cond_v2 module to cond_v2_impl, and creates a new cond_v2 module
that imports the cond_v2 method and the necessary extra imports. This is
useful for explicitly calling cond_v2 outside of control_flow_ops.cond.
PiperOrigin-RevId: 200778208
|
|
|
|
| |
PiperOrigin-RevId: 200777514
|
|
|
|
| |
PiperOrigin-RevId: 200774484
|
|
|
|
| |
PiperOrigin-RevId: 200771096
|
|
|
|
|
|
|
|
| |
Delete most print statements, use logging instead of print, and close files (to clear the "Unclosed file" warnings).
Normally this produces thousands of lines of output. Mostly noise.
PiperOrigin-RevId: 200769210
|
|
|
|
|
|
| |
mode.
PiperOrigin-RevId: 200768236
|
|
|
|
| |
PiperOrigin-RevId: 200766687
|
|
|
|
| |
PiperOrigin-RevId: 200764324
|
|
|
|
|
|
|
|
|
| |
but they were useful while debugging.
Open visibility of the low level gen_resource_variables_ops to compiler tests.
Fix bug in shape function of TPUReplicateInput for resource variables ? MergeInputHandleShapesAndTypes does not report shape mismatches.
PiperOrigin-RevId: 200756762
|
|
|
|
|
|
| |
fork-join parallelism (e.g., Eigen/ThreadpoolDevice or Shard).
PiperOrigin-RevId: 200756626
|
|
|
|
|
|
| |
TensorArraySizeV3 is a Placeholder.
PiperOrigin-RevId: 200755274
|
|
|
|
| |
PiperOrigin-RevId: 200753184
|
|
|
|
|
|
|
|
| |
AutoGraph-specific conversion logic: base converter classes, context objects, gensym. Largely, these are pulled out from impl and pyct.
This CL only adds the module - a future CL will replace existing implementations with these.
PiperOrigin-RevId: 200751782
|
|
|
|
|
|
|
| |
tensorrt optimizer (which requires access to the Device) to create a
transformed GraphDef.
PiperOrigin-RevId: 200751174
|
|
|
|
| |
PiperOrigin-RevId: 200751151
|
|
|
|
|
|
|
| |
HostTensorToLiteral function completely to prevent potential future misuse of
unnecessary memcpy.
PiperOrigin-RevId: 200750664
|
|
|
|
|
|
| |
variable restoration steps.
PiperOrigin-RevId: 200747752
|
|
|
|
|
|
| |
evaluation op.
PiperOrigin-RevId: 200747192
|
|
|
|
|
|
| |
HloInstruction::IdenticalSlowPath.
PiperOrigin-RevId: 200746735
|
|
|
|
| |
PiperOrigin-RevId: 200745718
|
|
|
|
|
|
|
| |
std::list is just hilariously inefficient and the postorder list creation has
been rewritten not to not depend on splicing anymore so there's no need for the
list. While there remove the old unused postorder list creation code.
PiperOrigin-RevId: 200743677
|
|
|
|
| |
PiperOrigin-RevId: 200742104
|
|
|
|
| |
PiperOrigin-RevId: 200741660
|
|
|
|
|
|
| |
to control inputs when converting nodes to NoOps.
PiperOrigin-RevId: 200740844
|
|
|
|
|
|
| |
Make resource_var.h more widely visible and add comment about the correct lock acquisition order if locking multiple variables.
PiperOrigin-RevId: 200736416
|
|
|
|
| |
PiperOrigin-RevId: 200736300
|
|
|
|
|
|
| |
and encapsulate_tpu_computations_pass, in order to detect errors earlier.
PiperOrigin-RevId: 200735435
|
|
|
|
| |
PiperOrigin-RevId: 200735157
|
|
|
|
| |
PiperOrigin-RevId: 200727545
|
|
|
|
| |
PiperOrigin-RevId: 200726967
|
|
|
|
|
|
|
|
| |
user-visible and which represent idioms not found in plain Python.
This CL only adds the module - a future CL will replace existing implementations with these.
PiperOrigin-RevId: 200712144
|
|
|
|
| |
PiperOrigin-RevId: 200702709
|
|
|
|
|
|
|
|
|
|
| |
When the demo app updates to use a backwards-incompatible change to the TFLite Java API at HEAD, it'll fail to build on the old release (which is missing the API change).
Using the nightly build means the demo app will use a relatively fresh TFLite
build with API changes, in addition to other improvements. The user may need to
pull the latest demo code to keep up.
PiperOrigin-RevId: 200672004
|
|
|
|
| |
PiperOrigin-RevId: 200667835
|
|
|
|
| |
PiperOrigin-RevId: 200667467
|
|
|
|
| |
PiperOrigin-RevId: 200662293
|
|
|
|
| |
PiperOrigin-RevId: 200661893
|
|
|
|
| |
PiperOrigin-RevId: 200661547
|