Searched refs:AllocateShapedBuffer (Results 1 – 3 of 3) sorted by relevance
53 Status AllocateShapedBuffer(DataType dtype, const xla::Shape& on_host_shape,
46 Status XlaTensor::AllocateShapedBuffer(DataType dtype, in AllocateShapedBuffer() function in tensorflow::XlaTensor
138 xla_tensor->AllocateShapedBuffer(device_tensor->dtype(), shape, client_, in CopyCPUTensorToDevice()