Searched refs:AllocateShapedBuffer (Results 1 – 4 of 4) sorted by relevance
51 Status AllocateShapedBuffer(DataType dtype, const xla::Shape& on_host_shape,
42 Status XlaTensor::AllocateShapedBuffer(DataType dtype, in AllocateShapedBuffer() function in tensorflow::XlaTensor
146 xla_tensor->AllocateShapedBuffer(device_tensor->dtype(), shape, client_, in CopyCPUTensorToDevice()
204 TF_RETURN_IF_ERROR(xla_output->AllocateShapedBuffer( in TpuDeviceToDeviceCopy()