Searched refs:XLA_ShapeIndex (Results 1 – 5 of 5) sorted by relevance
250 typedef struct XLA_ShapeIndex { struct253 } XLA_ShapeIndex; argument257 XLA_ShapeIndex* unowned_indices;266 XLA_ShapeIndex* aliased_indices;
61 XLA_ShapeIndex ToC(const xla::ShapeIndex& xla_shape);62 xla::ShapeIndex FromC(XLA_ShapeIndex* c_shape);63 void Free(XLA_ShapeIndex*);
317 XLA_ShapeIndex ToC(const xla::ShapeIndex& xla_shape) { in ToC()318 XLA_ShapeIndex c_shape; in ToC()327 xla::ShapeIndex FromC(XLA_ShapeIndex* c_shape) { in FromC()366 void Free(XLA_ShapeIndex* shape_index) { delete[] shape_index; } in Free()
316 XLA_ShapeIndex* array);
125 se_args[i]->unowned_indices = new XLA_ShapeIndex[unowned_indices.size()]; in ExecuteAsyncOnStream()