Searched defs:on_host_shape (Results 1 – 5 of 5) sorted by relevance
34 ShapedBuffer::ShapedBuffer(const Shape& on_host_shape, in ShapedBuffer()106 ScopedShapedBuffer::ScopedShapedBuffer(const Shape& on_host_shape, in ScopedShapedBuffer()
62 const Shape& on_host_shape() const { return on_host_shape_; } in on_host_shape() function
310 const Shape& on_host_shape, DeviceMemoryAllocator* allocator, in AllocateScopedShapedBuffer()
47 const xla::Shape& on_host_shape, in AllocateShapedBuffer()
169 const xla::Shape& on_host_shape, in XRTTupleAllocation()260 const xla::Shape& XRTTupleAllocation::on_host_shape() { return on_host_shape_; } in on_host_shape() function in tensorflow::XRTTupleAllocation