Home
last modified time | relevance | path

Searched refs:c_literal (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_transfer_manager.cc73 XLA_Literal c_literal; in TransferLiteralToDeviceAsync() local
74 ApiConverter::ToC(literal, &c_literal); in TransferLiteralToDeviceAsync()
83 &c_literal, &c_device_buffer, status.c_status); in TransferLiteralToDeviceAsync()
85 ApiConverter::Free(&c_literal); in TransferLiteralToDeviceAsync()
93 XLA_Literal c_literal; in TransferLiteralToInfeed() local
94 ApiConverter::ToC(literal, &c_literal); in TransferLiteralToInfeed()
98 manager_, tpu_executor->se_executor(), &c_literal, status.c_status); in TransferLiteralToInfeed()
100 ApiConverter::Free(&c_literal); in TransferLiteralToInfeed()
134 XLA_Literal c_literal; in TransferLiteralFromOutfeed() local
138 ApiConverter::ToC(literal, &c_literal); in TransferLiteralFromOutfeed()
[all …]
Dc_api_conversions.h66 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);
Dc_api_conversions.cc332 void ToC(const xla::LiteralSlice& literal, XLA_Literal* c_literal) { in ToC() argument
333 ApiConverter::ToC(literal.shape(), &c_literal->shape); in ToC()
335 c_literal->buffers = new char*[shapes.size()]; in ToC()
336 c_literal->sizes = new size_t[shapes.size()]; in ToC()
337 c_literal->count = shapes.size(); in ToC()
339 c_literal->buffers[i] = reinterpret_cast<char*>( in ToC()
341 c_literal->sizes[i] = literal.size_bytes(shapes[i].index); in ToC()
345 xla::MutableBorrowingLiteral FromC(XLA_Literal* c_literal) { in FromC() argument
346 xla::Shape shape = ApiConverter::FromC(&c_literal->shape); in FromC()
348 absl::MakeSpan(c_literal->buffers, c_literal->count), shape); in FromC()
[all …]
Dtpu_executor_c_api.h207 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);
/external/tensorflow/tensorflow/compiler/xla/tests/
Dscalar_computations_test.cc240 Literal c_literal = LiteralUtil::CreateR0<float>(0.5f); in XLA_TEST_F() local
247 client_->TransferToServer(c_literal).ConsumeValueOrDie(); in XLA_TEST_F()
251 XlaOp c = Parameter(&builder, 2, c_literal.shape(), "c"); in XLA_TEST_F()
/external/mesa3d/docs/relnotes/
D20.1.0.rst3241 - intel/tools: Do not print type/qualifiers/name for c_literal
3803 - intel/tools: Allow i965_disasm to disassemble c_literal input type