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

  /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/kernels/
xrt_execute_op.cc 225 auto scoped_buffer = run_result.ConsumeValueOrDie(); local
226 auto shaped_buffer = scoped_buffer.release();
  /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();

Completed in 315 milliseconds