Searched refs:XLA_Literal (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | c_api_conversions.h | 66 void ToC(const xla::LiteralSlice& literal, XLA_Literal* c_literal);
67 xla::MutableBorrowingLiteral FromC(XLA_Literal* c_literal);
68 void Free(XLA_Literal* c_literal);
81 void ToC(const xla::LiteralSlice& literal, XLA_Literal* c_literal);
82 xla::MutableBorrowingLiteral FromC(XLA_Literal* c_literal);
83 void Free(XLA_Literal* c_literal);
|
| D | tpu_transfer_manager.cc | 73 XLA_Literal c_literal; in TransferLiteralToDeviceAsync()
93 XLA_Literal c_literal; in TransferLiteralToInfeed()
134 XLA_Literal c_literal; in TransferLiteralFromOutfeed()
199 XLA_Literal c_literal; in TransferLiteralFromDevice()
299 XLA_Literal c_literal; in LinearizeToBuffers()
|
| D | tpu_executor_c_api.h | 183 XLA_TransferManager* manager, SE_Stream* stream, XLA_Literal* literal,
187 XLA_ShapedBuffer* device_buffer, XLA_Literal* literal,
207 XLA_TransferManager* manager, XLA_Literal* c_literal, char*** buffers_array,
213 XLA_Literal* c_literal,
223 XLA_Shape* shape /*deprecated*/, XLA_Literal* c_literal, TF_Status* status);
|
| D | c_api_decl.h | 238 typedef struct XLA_Literal { struct
243 } XLA_Literal; argument
|
| D | c_api_conversions.cc | 332 void ToC(const xla::LiteralSlice& literal, XLA_Literal* c_literal) { in ToC()
345 xla::MutableBorrowingLiteral FromC(XLA_Literal* c_literal) { in FromC()
369 void Free(XLA_Literal* c_literal) { in Free()
|