Searched refs:GlobalDataToShapedBuffer (Results 1 – 5 of 5) sorted by relevance
63 StatusOr<const ShapedBuffer*> GlobalDataToShapedBuffer(
189 StatusOr<const ShapedBuffer*> LocalService::GlobalDataToShapedBuffer( in GlobalDataToShapedBuffer() function in xla::LocalService
286 StatusOr<const ShapedBuffer*> LocalClient::GlobalDataToShapedBuffer( in GlobalDataToShapedBuffer() function in xla::LocalClient288 return local_service_->GlobalDataToShapedBuffer(data, replica_number); in GlobalDataToShapedBuffer()
142 StatusOr<const ShapedBuffer*> GlobalDataToShapedBuffer(
217 client->GlobalDataToShapedBuffer(data->handle(), /*device_ordinal=*/0) in ReplayComputation()