aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/core/grappler/costs
Commit message (Collapse)AuthorAge
* Change LOG(WARNING) to VLOG(1) in utilsGravatar Peter Ma2018-10-09
| | | | PiperOrigin-RevId: 216369081
* Refactor CalculateOutputSize() from VirtualScheduler protected member ↵Gravatar Peter Ma2018-10-08
| | | | | | function to utils; Refactor EstimateSize() from memory_optimizer.cc to utils; some small changes for readability improvement PiperOrigin-RevId: 216307257
* Consolidate device parameter arguments into a shared DeviceInfo structGravatar A. Unique TensorFlower2018-10-08
| | | | PiperOrigin-RevId: 216280197
* Fix bug in Grappler constant folding: The logic detecting full reductions ↵Gravatar A. Unique TensorFlower2018-10-04
| | | | | | | | 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
* PinToHostOptimizer: Refactored code. Update blacklist. Added recursive ↵Gravatar A. Unique TensorFlower2018-10-03
| | | | | | lookback for Identity op. This fixes many performance regressions. PiperOrigin-RevId: 215662393
* Support shape_invariants in while_v2. Note that this arg is temporary and ↵Gravatar Saurabh Saxena2018-10-02
| | | | | | | | 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
* Merge pull request #21000 from ↵Gravatar TensorFlower Gardener2018-09-19
|\ | | | | | | | | | | ROCmSoftwarePlatform:upstream-staging-gpu-common-runtime-1 PiperOrigin-RevId: 213653830
* | [Grappler] s/std::string/string/Gravatar James Keeling2018-09-14
| | | | | | | | | | | | string and std::string are not necessarily the same thing in TF, but this code assumed that they are. PiperOrigin-RevId: 212952877
* | Improve static shape inference in grappler by propagating tensors_as_shapes ↵Gravatar Doe Hyun Yoon2018-09-12
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | 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
* | Replace Placeholder with Const to GrapplerFunctionItem for function shape ↵Gravatar Doe Hyun Yoon2018-09-06
| | | | | | | | | | | | inference if possible. PiperOrigin-RevId: 211821596
| * Rename CUDA GPU ID to platform GPU IDGravatar Wen-Heng (Jack) Chung2018-08-31
|/ | | | | Rename CUDA GPU ID to platform GPU ID so the notion is applicable on both CUDA and ROCm platform.
* Directly import tensor.proto.h (the transitive import will be removed from ↵Gravatar Eugene Brevdo2018-08-23
| | | | | | | | | | 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
* Sorted the per-device summary printout with device names to improve readability.Gravatar A. Unique TensorFlower2018-08-23
| | | | PiperOrigin-RevId: 210007888
* Add graphdef version number to GrapplerFunctionItem.Gravatar Doe Hyun Yoon2018-08-23
| | | | | | | | 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
* Replaced calls to tensorflow::StringPiece::ToString with string conversions.Gravatar A. Unique TensorFlower2018-08-22
| | | | | | | | 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
* Small fix to MaybeGetMinimumShape() in op_level_cost_estimator.Gravatar Doe Hyun Yoon2018-08-15
| | | | | | | | | | | | | | | (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
* Add two counters in Costs Struct for number of ops processed/predicted in ↵Gravatar Peter Ma2018-08-10
| | | | | | total, and number of ops predicted with unknown shapes PiperOrigin-RevId: 208274158
* Minor code cleanup in virtual_scheduler.Gravatar Max Galkin2018-08-03
| | | | PiperOrigin-RevId: 207334214
* Explicitly cast the types of a few variables in VLOG statements to avoid an ↵Gravatar A. Unique TensorFlower2018-08-02
| | | | | | issue where the compiler isn't sure of the type when building for arm64 computers. PiperOrigin-RevId: 207151595
* Move GetDeviceStates() and GetNodeState() functions from protected to ↵Gravatar A. Unique TensorFlower2018-08-01
| | | | | | public, so that the user can have access to more detailed results from VirtualScheduler. PiperOrigin-RevId: 206986812
* Added QuantizedMatMul to OpLevelCostEstimator.Gravatar A. Unique TensorFlower2018-07-24
| | | | PiperOrigin-RevId: 205918819
* Tabularized the VLOGs printing per-op execution times to make it easier to ↵Gravatar A. Unique TensorFlower2018-07-24
| | | | | | see what ops take much time. PiperOrigin-RevId: 205913222
* Update Grappler to use existing functions for retrieving a node'sGravatar A. Unique TensorFlower2018-07-20
| | | | | | name and position. PiperOrigin-RevId: 205465354
* Function shape inference for Grappler.Gravatar A. Unique TensorFlower2018-07-10
| | | | PiperOrigin-RevId: 203962286
* Support filter format for FusedConv2DBiasActivation.Gravatar Yao Zhang2018-06-20
| | | | PiperOrigin-RevId: 201454730
* Merge changes from github.Gravatar Akshay Modi2018-06-18
| | | | PiperOrigin-RevId: 201110240
* Automated g4 rollback of changelist 201011811Gravatar Akshay Modi2018-06-18
| | | | PiperOrigin-RevId: 201033171
* Merge changes from github.Gravatar Akshay Modi2018-06-18
| | | | PiperOrigin-RevId: 201011811
* Faster TopoQueue in graph_properties.Gravatar Max Galkin2018-06-15
| | | | PiperOrigin-RevId: 200791799
* Set shapes and types to queue ops, if not set by enqueue ops.Gravatar A. Unique TensorFlower2018-06-15
| | | | PiperOrigin-RevId: 200764324
* CostGraphDef has been modified to keep track of the accuracy of the cost ↵Gravatar A. Unique TensorFlower2018-06-11
| | | | | | estimation. PiperOrigin-RevId: 200078367
* Estimate Squeeze cost in the same way as Reshape.Gravatar Max Galkin2018-06-06
| | | | PiperOrigin-RevId: 199531069
* Resubmitting CL 196349902: Adding cuDNN header dependency to targets that ↵Gravatar A. Unique TensorFlower2018-06-01
| | | | | | include the cuDNN header file. PiperOrigin-RevId: 198869605
* Putting stubs for function shape inference interfaceGravatar A. Unique TensorFlower2018-05-31
| | | | PiperOrigin-RevId: 198794845
* Merge changes from github.Gravatar Yifei Feng2018-05-24
| | | | | | | 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
* Added support for strided slicing of symbolic shapesGravatar Benoit Steiner2018-05-14
| | | | PiperOrigin-RevId: 196558466
* Automated g4 rollback of changelist 196349902Gravatar A. Unique TensorFlower2018-05-12
| | | | PiperOrigin-RevId: 196387391
* Adding cuDNN header dependency to targets that include the cuDNN header file.Gravatar A. Unique TensorFlower2018-05-11
| | | | PiperOrigin-RevId: 196349902
* Expose MaybeGetMinimumShape for use in cost estimators other than ↵Gravatar A. Unique TensorFlower2018-05-11
| | | | | | OpLevelCostEstimator. PiperOrigin-RevId: 196315239
* Fix a dropped line in the DepthwiseConv2dNative modelGravatar A. Unique TensorFlower2018-05-09
| | | | PiperOrigin-RevId: 195900021
* Add cost model of depthwiseConv2dNative. Tensorflow computes depthwise ↵Gravatar A. Unique TensorFlower2018-05-08
| | | | | | separable convolutions as depthwiseConv2dNative followed by 1x1 Conv2D PiperOrigin-RevId: 195838887
* GuaranteeConst is a NoOp for the op_level_cost_estiamtor.Gravatar Max Galkin2018-05-05
| | | | PiperOrigin-RevId: 195501990
* Handle negative values when slicing symbolic shapesGravatar Benoit Steiner2018-05-02
| | | | PiperOrigin-RevId: 195176133
* Added support for packing of symbolic shapesGravatar Benoit Steiner2018-05-02
| | | | PiperOrigin-RevId: 195137239
* Optimized the analysis of rank and size operations.Gravatar Benoit Steiner2018-05-02
| | | | PiperOrigin-RevId: 195108832
* Avoid making a copy of the graph needlesslyGravatar Benoit Steiner2018-05-01
| | | | PiperOrigin-RevId: 195017837
* Fixed some outdated commentsGravatar Benoit Steiner2018-05-01
| | | | PiperOrigin-RevId: 195006088
* Merge changes from github.Gravatar Patrick Nguyen2018-05-01
| | | | PiperOrigin-RevId: 194997009
* Simplified shape inference.Gravatar Benoit Steiner2018-05-01
| | | | PiperOrigin-RevId: 194975603
* Format header guards under tensorflow/core/grappler.Gravatar A. Unique TensorFlower2018-04-26
| | | | PiperOrigin-RevId: 194387041