Home
last modified time | relevance | path

Searched refs:XLA_Tile (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_decl.h198 typedef struct XLA_Tile { struct
200 } XLA_Tile; typedef
204 XLA_Tile* heap; // owned
205 XLA_Tile inlined[TPU_C_API_MAX_INLINED];
Dc_api_conversions.h56 xla::Tile FromC(const XLA_Tile* c_tile);
57 void ToC(const xla::Tile& xla_tile, XLA_Tile* c_tile);
58 void Free(XLA_Tile* c_tile);
Dc_api_conversions.cc178 XLA_Tile* c_tiles; in CopyVector()
180 dst->heap = new XLA_Tile[dst->size]; in CopyVector()
282 const XLA_Tile* c_tiles = c_layout->tiles.size > TPU_C_API_MAX_INLINED in FromC()
301 void ToC(const xla::Tile& tile, XLA_Tile* c_tile) { in ToC()
305 xla::Tile FromC(const XLA_Tile* c_tile) { in FromC()
311 void Free(XLA_Tile* c_tile) { in Free()