Searched refs:GlobalDataToShapedBuffer (Results 1 – 5 of 5) sorted by relevance
64 StatusOr<const ShapedBuffer*> GlobalDataToShapedBuffer(
196 StatusOr<const ShapedBuffer*> LocalService::GlobalDataToShapedBuffer( in GlobalDataToShapedBuffer() function in xla::LocalService
166 StatusOr<const ShapedBuffer*> GlobalDataToShapedBuffer(
423 StatusOr<const ShapedBuffer*> LocalClient::GlobalDataToShapedBuffer( in GlobalDataToShapedBuffer() function in xla::LocalClient425 return local_service_->GlobalDataToShapedBuffer(data, replica_number); in GlobalDataToShapedBuffer()
249 client->GlobalDataToShapedBuffer(data->handle(), /*device_ordinal=*/0) in ReplayComputation()