aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/xla/sparse_index_array.h
Commit message (Collapse)AuthorAge
* Change headers to directly include absl::Span, and clean up the buildGravatar Tim Shen2018-08-30
| | | | | | dependencies as well. PiperOrigin-RevId: 211038094
* [XLA] Rename all (Mutable)ArraySlice to absl::Span.Gravatar Tim Shen2018-08-30
| | | | PiperOrigin-RevId: 210998142
* Remove (Mutable)ArraySlice implementation and alias them to absl::Span.Gravatar Tim Shen2018-08-30
| | | | | | | | There are several API migrations happening: * ArraySlice's sub-slice constructor => .subspan * MutableArraySlice's container pointer constructor => absl::MakeSpan PiperOrigin-RevId: 210946124
* Merged commit includes the following changes:Gravatar Yifei Feng2018-08-21
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | 209663919 by yifeif<yifeif@google.com>: Internal change. -- 209663914 by amitpatankar<amitpatankar@google.com>: Fix the topk_op_test for numpy>1.15. -- 209660476 by jdduke<jdduke@google.com>: Fix model lifetime for TensorFlow Lite C# bindings Ensure the model's existence for the duration of the interpreter, as per API requirements. -- 209655960 by scottzhu<scottzhu@google.com>: Unify RNN Cell interface between TF and Keras. -- 209655731 by A. Unique TensorFlower<gardener@tensorflow.org>: Added tests for PredictionOps and PartitionExamplesOps -- 209655291 by nolivia<nolivia@google.com>: adding rate class so that we can save global_step/sec using tf.contrib.summary. The function takes the rate in relation to any tensors provided that the numerator and denominator are broadcastable and have dtypes that can be cast to float64 -- 209654655 by kramerb<kramerb@google.com>: [XLA] Switch from tensorflow::gtl::InlinedVector to absl::InlinedVector This one comes with extra goodies like a move constructor. -- 209653851 by A. Unique TensorFlower<gardener@tensorflow.org>: Internal build specification change -- PiperOrigin-RevId: 209663919
* Remove THIRD_PARTY_ from #include guardsGravatar Sanjoy Das2018-01-24
| | | | | | They don't make sense in the open source repository. PiperOrigin-RevId: 183140889
* [XLA:TPU] Initial HLO parser/stringifier support for sparse formatsGravatar A. Unique TensorFlower2018-01-12
| | | | | | | | | | | | | | | | | | | | | | | | | | | | | | | - Add methods for manipulating sparse literals to xla::Literal - Make LayoutUtil::HumanString handle sparse layouts - Make ShapeUtil::ParseShape handle sparse shapes - Syntax for shapes has changed: - Old way of expressing layouts still works, e.g. f32[1,2,3]{2,1,0} - Can now make dense format explicit: f32[1,2,3]dense{2,1,0} - Can express sparse layouts; the max_sparse_elements value is in the braces, e.g.: f32[1,2,3]sparse{10} - The shape should not include braces for the layout if the shape is scalar; e.g. f32[]{} is not valid shape syntax. - The shape should not include braces for the layout if the shape is a dense rank-1 array; e.g. f32[10]{0} is not valid shape syntax - Sparse literals use a dictionary-liky syntax, e.g.: f32[2,3,4]sparse{10} {[0,1,2]: 10, [1,2,3]: 11} - For rank-1 sparse arrays, the square brackets around indices may be omitted, e.g.: f32[100]sparse{10} {5: 10, 20: 30} PiperOrigin-RevId: 181813837
* [XLA] Initial sparse layout supportGravatar A. Unique TensorFlower2018-01-08
Adds SparseIndexArray and support methods to Literal. SparseIndexArray manages the array of sparse indices and is exposed by sparse Literals. Also adds HloSupportChecker classes for CPU and GPU. This will be run as the first HloPass during compilation, and verifies that the graph is supported by the backend. Currently only verifies shapes, and that the layout is not sparse since no backend supports sparse layouts yet. PiperOrigin-RevId: 181244401