| Commit message (Collapse) | Author | Age |
|
|
|
| |
PiperOrigin-RevId: 216369081
|
|
|
|
|
|
| |
function to utils; Refactor EstimateSize() from memory_optimizer.cc to utils; some small changes for readability improvement
PiperOrigin-RevId: 216307257
|
|
|
|
| |
PiperOrigin-RevId: 216280197
|
|
|
|
|
|
|
|
| |
was flawed. Added better test coverage.
Also added a extra test for a related symbolic shape inference operation that I first suspected to be broken.
PiperOrigin-RevId: 215812753
|
|
|
|
|
|
| |
lookback for Identity op. This fixes many performance regressions.
PiperOrigin-RevId: 215662393
|
|
|
|
|
|
|
|
| |
may be replaced by automatic shape inference in TF 2.0 (or before).
Add a output_shapes attr to While op to allow output shapes to be different from the incoming loop_vars.
PiperOrigin-RevId: 215446737
|
|\
| |
| |
| |
| |
| | |
ROCmSoftwarePlatform:upstream-staging-gpu-common-runtime-1
PiperOrigin-RevId: 213653830
|
| |
| |
| |
| |
| |
| | |
string and std::string are not necessarily the same thing in TF, but this code assumed that they are.
PiperOrigin-RevId: 212952877
|
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| | |
better:
Currently, static shape inference propagates shapes of tensors, but in some cases, we do need values; for this, we use input_tensors (from Const input tensor) and input_tensors_as_shapes and output_tensors_as_shapes (these are ShapeHandle format, but has values, currently only for 1D vector).
This CL enhances propagation of input_tensors_as_shapes and output_tensors_as_shapes to improve static shape inference.
(1) forward scalar Const as input_tensors_as_shapes (currently, only 1D vector),
(2) export input_tensors_as_shapes, output const tensor, and output_tensors_as_shapes to the values of inferred input/output TensorProperties (currently, only input const tensors are exported as values),
(3) use input_tensors_as_shapes as Const tensor to function input (currently, only Const tensors),
(4) forward input_tensors_as_shapes to output_tensors_as_shapes for Identity op,
(5) when Pack op concats scalar values to form output_tensors_as_shapes, currently it uses only input_tensors (from Const input tensors), but this CL change Pack to use input_tensors_as_shapes as well.
PiperOrigin-RevId: 212696959
|
| |
| |
| |
| |
| |
| | |
inference if possible.
PiperOrigin-RevId: 211821596
|
|/
|
|
|
| |
Rename CUDA GPU ID to platform GPU ID so the notion is applicable on both CUDA
and ROCm platform.
|
|
|
|
|
|
|
|
|
|
| |
tensor.h soon)
We plan to remove the import variant.h from tensor.h; and variant.h brings in a lot
of transitive imports (including protos like tensor.proto.h). To prepare, we're
updating folks who this will break.
PiperOrigin-RevId: 210043667
|
|
|
|
| |
PiperOrigin-RevId: 210007888
|
|
|
|
|
|
|
|
| |
This solves te problem when passing a scalar tensor to function op input, as
Placeholer shape inference outputs unknown shape for scalar if graphdef version
is < 24.
PiperOrigin-RevId: 210007276
|
|
|
|
|
|
|
|
| |
That is, instances of sp.ToString() are replaced with string(sp).
This will allow tensorflow::StringPiece::ToString to be removed, which is necessary before it can be replaced with absl::string_view.
PiperOrigin-RevId: 209806694
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| |
(1) previously, it set unknown shape flag for scalar input, but now it
returns TensorShapeProto with rank equal to the expected and all dims set to 1,
and unknown shape flag is not set.
(2) Also, fixed a bug; when a rank is known, but dim_size() < rank (note that
dim_size() may be non-zero), we previously called add_dim() with dim 1 rank
times, which then makes dim_size() is incremented by rank, but we expect
dim_size() equal to rank.
(3) Added test for MaybeGetMinimumShape().
PiperOrigin-RevId: 208845501
|
|
|
|
|
|
| |
total, and number of ops predicted with unknown shapes
PiperOrigin-RevId: 208274158
|
|
|
|
| |
PiperOrigin-RevId: 207334214
|
|
|
|
|
|
| |
issue where the compiler isn't sure of the type when building for arm64 computers.
PiperOrigin-RevId: 207151595
|
|
|
|
|
|
| |
public, so that the user can have access to more detailed results from VirtualScheduler.
PiperOrigin-RevId: 206986812
|
|
|
|
| |
PiperOrigin-RevId: 205918819
|
|
|
|
|
|
| |
see what ops take much time.
PiperOrigin-RevId: 205913222
|
|
|
|
|
|
| |
name and position.
PiperOrigin-RevId: 205465354
|
|
|
|
| |
PiperOrigin-RevId: 203962286
|
|
|
|
| |
PiperOrigin-RevId: 201454730
|
|
|
|
| |
PiperOrigin-RevId: 201110240
|
|
|
|
| |
PiperOrigin-RevId: 201033171
|
|
|
|
| |
PiperOrigin-RevId: 201011811
|
|
|
|
| |
PiperOrigin-RevId: 200791799
|
|
|
|
| |
PiperOrigin-RevId: 200764324
|
|
|
|
|
|
| |
estimation.
PiperOrigin-RevId: 200078367
|
|
|
|
| |
PiperOrigin-RevId: 199531069
|
|
|
|
|
|
| |
include the cuDNN header file.
PiperOrigin-RevId: 198869605
|
|
|
|
| |
PiperOrigin-RevId: 198794845
|
|
|
|
|
|
|
| |
Revert #18413. Too many internal test failures due to the name scope change caused by this change.
Revert #18192. Cannot use re2::StringPiece internally. Need alternative for set call. Will pull and clean this up in a separate change.
PiperOrigin-RevId: 197991247
|
|
|
|
| |
PiperOrigin-RevId: 196558466
|
|
|
|
| |
PiperOrigin-RevId: 196387391
|
|
|
|
| |
PiperOrigin-RevId: 196349902
|
|
|
|
|
|
| |
OpLevelCostEstimator.
PiperOrigin-RevId: 196315239
|
|
|
|
| |
PiperOrigin-RevId: 195900021
|
|
|
|
|
|
| |
separable convolutions as depthwiseConv2dNative followed by 1x1 Conv2D
PiperOrigin-RevId: 195838887
|
|
|
|
| |
PiperOrigin-RevId: 195501990
|
|
|
|
| |
PiperOrigin-RevId: 195176133
|
|
|
|
| |
PiperOrigin-RevId: 195137239
|
|
|
|
| |
PiperOrigin-RevId: 195108832
|
|
|
|
| |
PiperOrigin-RevId: 195017837
|
|
|
|
| |
PiperOrigin-RevId: 195006088
|
|
|
|
| |
PiperOrigin-RevId: 194997009
|
|
|
|
| |
PiperOrigin-RevId: 194975603
|
|
|
|
| |
PiperOrigin-RevId: 194387041
|