HomeSort by relevance Sort by last modified time
    Searched refs:scoped_buffer (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/client/
local_client.cc 268 TF_ASSIGN_OR_RETURN(auto scoped_buffer,
274 stream.get(), literal, scoped_buffer));
275 return std::move(scoped_buffer);
  /external/tensorflow/tensorflow/compiler/xla/service/
shaped_buffer_test.cc 40 auto scoped_buffer = absl::make_unique<xla::ScopedShapedBuffer>( local
42 std::unique_ptr<xla::ShapedBuffer> buffer = std::move(scoped_buffer);
  /external/tensorflow/tensorflow/compiler/xrt/
xrt_state.cc 189 std::unique_ptr<xla::ScopedShapedBuffer> scoped_buffer; local
191 backend, device_ordinal, literal.shape(), &scoped_buffer));
194 stream.get(), literal, *scoped_buffer));
200 auto shaped_buffer = scoped_buffer->release();
  /external/tensorflow/tensorflow/compiler/xrt/kernels/
xrt_execute_op.cc 225 auto scoped_buffer = run_result.ConsumeValueOrDie(); local
226 auto shaped_buffer = scoped_buffer.release();

Completed in 248 milliseconds