Searched defs:shape_index (Results 1 – 8 of 8) sorted by relevance
140 const ShapeIndex& shape_index, in GetTypedIrValue()161 const ShapeIndex& shape_index) { in BindHloToIrValue()198 const ShapeIndex& shape_index) { in GetIrArray()
746 Piece& piece(const ShapeIndex& shape_index) { in piece()749 const Piece& piece(const ShapeIndex& shape_index) const { in piece()846 const ShapeIndex& shape_index) const { in data()852 const ShapeIndex& shape_index) { in data()858 const ShapeIndex& shape_index) const { in Get()870 const ShapeIndex& shape_index, NativeT value) { in Set()1116 const ShapeIndex& shape_index) const { in GetSparseElement()1125 const ShapeIndex& shape_index) { in AppendSparseElement()
179 SparseIndexArray* Literal::sparse_indices(const ShapeIndex& shape_index) { in sparse_indices()1008 void Literal::SortSparseElements(const ShapeIndex& shape_index) { in SortSparseElements()1074 void ToStringHelper(const Literal& literal, const ShapeIndex& shape_index, in ToStringHelper()1833 void* Literal::untyped_data(const ShapeIndex& shape_index) { in untyped_data()
57 std::ostream& operator<<(std::ostream& out, const ShapeIndex& shape_index) { in operator <<()62 std::ostream& operator<<(std::ostream& out, const ShapeIndexView& shape_index) { in operator <<()
138 [&output_rank](const Shape& subshape, const ShapeIndex& shape_index) { in EffectivelyUnary()
179 repeated int64 shape_index = 3; field
3407 const ShapeIndex& shape_index) { in RelayoutConstant()
126 [&](const Shape& subshape, const ShapeIndex& shape_index) { in ConvertType()