MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / ShapedBuffer

Method ShapedBuffer

tensorflow/compiler/xla/service/shaped_buffer.cc:34–40  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

32namespace xla {
33
34ShapedBuffer::ShapedBuffer(Shape on_host_shape, Shape on_device_shape,
35 const se::Platform* platform, int device_ordinal)
36 : on_host_shape_(std::move(on_host_shape)),
37 on_device_shape_(std::move(on_device_shape)),
38 platform_(platform),
39 device_ordinal_(device_ordinal),
40 buffers_(&on_device_shape_) {}
41
42ShapedBuffer::ShapedBuffer(ShapedBuffer&& s)
43 : on_host_shape_(std::move(s.on_host_shape_)),

Callers

nothing calls this directly

Calls 1

replace_shape_ptrMethod · 0.80

Tested by

no test coverage detected